gafime-rocm 0.4.7__tar.gz

gafime_rocm-0.4.7/MANIFEST.in

@@ -0,0 +1,7 @@
+include README.md
+recursive-include gafime_rocm *
+recursive-include gafime _dummy.c
+recursive-include src/common interfaces.h
+recursive-include src/rocm kernels.hip
+global-exclude *.py[cod]
+global-exclude __pycache__

gafime_rocm-0.4.7/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.4
+Name: gafime-rocm
+Version: 0.4.7
+Summary: AMD ROCm/HIP runtime payload for GAFIME
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: gafime==0.4.7
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# gafime-rocm
+
+Vendor GPU runtime payload for GAFIME 0.4.7.
+
+This package is generated from the GAFIME source tree during CI and carries
+only the ROCM native runtime payload. Install the base package
+with `gafime`; use this package only for the matching GPU runtime.

gafime_rocm-0.4.7/README.md

@@ -0,0 +1,7 @@
+# gafime-rocm
+
+Vendor GPU runtime payload for GAFIME 0.4.7.
+
+This package is generated from the GAFIME source tree during CI and carries
+only the ROCM native runtime payload. Install the base package
+with `gafime`; use this package only for the matching GPU runtime.

gafime_rocm-0.4.7/gafime/_dummy.c

@@ -0,0 +1,11 @@
+#include <Python.h>
+static struct PyModuleDef dummy_module = {
+    PyModuleDef_HEAD_INIT,
+    "_native",
+    "Dummy extension block",
+    -1,
+    NULL, NULL, NULL, NULL, NULL
+};
+PyMODINIT_FUNC PyInit__native(void) {
+    return PyModule_Create(&dummy_module);
+}

gafime_rocm-0.4.7/gafime_rocm/__init__.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+__version__ = "0.4.7"
+
+
+def package_dir() -> Path:
+    return Path(__file__).resolve().parent
+
+
+def library_candidates() -> list[Path]:
+    base = package_dir()
+    return [
+        base / "gafime_rocm.dll",
+        base / "libgafime_rocm.so",
+        base / "gafime_rocm.so",
+        base / "gafime_rocm.pyd",
+    ]

gafime_rocm-0.4.7/gafime_rocm.egg-info/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.4
+Name: gafime-rocm
+Version: 0.4.7
+Summary: AMD ROCm/HIP runtime payload for GAFIME
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: gafime==0.4.7
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# gafime-rocm
+
+Vendor GPU runtime payload for GAFIME 0.4.7.
+
+This package is generated from the GAFIME source tree during CI and carries
+only the ROCM native runtime payload. Install the base package
+with `gafime`; use this package only for the matching GPU runtime.

gafime_rocm-0.4.7/gafime_rocm.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+MANIFEST.in
+README.md
+pyproject.toml
+setup.py
+/home/runner/work/GAFIME/GAFIME/payload-src/gafime-rocm/gafime/_dummy.c
+gafime/_dummy.c
+gafime_rocm/__init__.py
+gafime_rocm.egg-info/PKG-INFO
+gafime_rocm.egg-info/SOURCES.txt
+gafime_rocm.egg-info/dependency_links.txt
+gafime_rocm.egg-info/requires.txt
+gafime_rocm.egg-info/top_level.txt
+src/common/interfaces.h
+src/rocm/kernels.hip

gafime_rocm-0.4.7/gafime_rocm.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

gafime_rocm-0.4.7/gafime_rocm.egg-info/requires.txt

@@ -0,0 +1 @@
+gafime==0.4.7

gafime_rocm-0.4.7/gafime_rocm.egg-info/top_level.txt

@@ -0,0 +1 @@
+gafime_rocm

gafime_rocm-0.4.7/pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools>=77", "wheel"]
+build-backend = "setuptools.build_meta"

gafime_rocm-0.4.7/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

gafime_rocm-0.4.7/setup.py

@@ -0,0 +1,159 @@
+from __future__ import annotations
+
+import os
+import platform
+import shutil
+import subprocess
+import sys
+from pathlib import Path
+
+from setuptools import Extension, setup
+from setuptools.command.build_ext import build_ext
+
+
+VERSION = "0.4.7"
+ROOT = Path(__file__).resolve().parent
+
+
+class RocmPayloadBuildExt(build_ext):
+    def run(self):
+        package_dir = Path(self.build_lib) / "gafime_rocm"
+        package_dir.mkdir(parents=True, exist_ok=True)
+        self.output_dir = package_dir
+        self.build_rocm_backend()
+        super().run()
+
+    def build_rocm_backend(self) -> None:
+        if sys.platform not in {"linux", "win32"}:
+            raise RuntimeError("gafime-rocm currently supports Linux/Windows x86_64 targets.")
+        machine = platform.machine().lower()
+        if machine in {"aarch64", "arm64"} or machine.startswith("arm"):
+            raise RuntimeError(f"gafime-rocm does not support ARM target {platform.machine()}.")
+
+        hipcc = shutil.which("hipcc")
+        if not hipcc:
+            raise RuntimeError("hipcc was not found. Install ROCm/HIP to build gafime-rocm.")
+
+        src_dir = ROOT / "src"
+        rocm_source = src_dir / "rocm" / "kernels.hip"
+        output_file = self.output_dir / ("gafime_rocm.dll" if sys.platform == "win32" else "libgafime_rocm.so")
+        arch_env = os.environ.get("GAFIME_ROCM_ARCHS")
+        if arch_env:
+            arch_mode = arch_env.strip().lower().replace("_", "-")
+            if arch_mode in {"release", "package", "wheel", "release-wheel"}:
+                archs = self._windows_release_rocm_archs() if sys.platform == "win32" else self._linux_release_rocm_archs()
+            elif arch_mode in {"linux-release", "linux-wheel"}:
+                archs = self._linux_release_rocm_archs()
+            elif arch_mode in {"windows-release", "windows-wheel"}:
+                archs = self._windows_release_rocm_archs()
+            else:
+                archs = [arch.strip() for arch in arch_env.replace(";", ",").replace(" ", ",").split(",") if arch.strip()]
+        else:
+            archs = self._detect_rocm_archs()
+        if not archs:
+            raise RuntimeError(
+                "Unable to detect ROCm/HIP offload architecture. "
+                "Set GAFIME_ROCM_ARCHS explicitly, for example GAFIME_ROCM_ARCHS=<rocm-offload-target>."
+            )
+        arch_flags = [f"--offload-arch={arch}" for arch in archs]
+
+        cmd = [
+            hipcc,
+            *arch_flags,
+            "-O3",
+            "--shared",
+            "-Wno-unused-result",
+            "-DGAFIME_BUILDING_DLL",
+            "-I",
+            str(src_dir / "common"),
+            "-o",
+            str(output_file),
+            str(rocm_source),
+        ]
+        if sys.platform != "win32":
+            cmd.insert(cmd.index("-Wno-unused-result"), "-fPIC")
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        if result.returncode != 0:
+            raise RuntimeError(f"ROCm/HIP build failed\nSTDOUT:\n{result.stdout}\nSTDERR:\n{result.stderr}")
+
+    @staticmethod
+    def _linux_release_rocm_archs() -> list[str]:
+        # ROCm does not provide a single NVIDIA-PTX-like forward-compatible
+        # code object for every AMD GPU. Release wheels therefore carry a
+        # package-policy target set covering current ROCm 7.x client, APU, and
+        # datacenter families instead of baking in one developer machine target.
+        return [
+            "gfx90a",
+            "gfx942",
+            "gfx950",
+            "gfx1030",
+            "gfx1031",
+            "gfx1032",
+            "gfx1100",
+            "gfx1101",
+            "gfx1102",
+            "gfx1150",
+            "gfx1151",
+            "gfx1200",
+            "gfx1201",
+        ]
+
+    @staticmethod
+    def _windows_release_rocm_archs() -> list[str]:
+        # Windows HIP SDK publishes a narrower officially supported set than
+        # Linux ROCm. Keep this list aligned with AMD's Windows HIP SDK support
+        # table so CI does not feed datacenter/Linux-only code objects to the
+        # Windows installer toolchain.
+        return [
+            "gfx1030",
+            "gfx1031",
+            "gfx1032",
+            "gfx1100",
+            "gfx1101",
+            "gfx1102",
+            "gfx1151",
+            "gfx1200",
+            "gfx1201",
+        ]
+
+    @staticmethod
+    def _detect_rocm_archs() -> list[str]:
+        enumerator = shutil.which("rocm_agent_enumerator")
+        if enumerator:
+            try:
+                result = subprocess.run([enumerator], capture_output=True, text=True, check=False, timeout=10)
+                archs: list[str] = []
+                for line in result.stdout.splitlines():
+                    arch = line.strip()
+                    if arch.startswith("gfx") and arch not in archs:
+                        archs.append(arch)
+                if archs:
+                    return archs
+            except Exception:
+                pass
+        return []
+
+
+setup_kwargs = dict(
+    name="gafime-rocm",
+    version=VERSION,
+    description="AMD ROCm/HIP runtime payload for GAFIME",
+    long_description=(ROOT / "README.md").read_text(encoding="utf-8"),
+    long_description_content_type="text/markdown",
+    packages=["gafime_rocm"],
+    package_data={"gafime_rocm": ["*.so", "*.dll", "*.pyd"]},
+    include_package_data=False,
+    install_requires=[f"gafime=={VERSION}"],
+    python_requires=">=3.10",
+    ext_modules=[Extension("gafime_rocm._native", sources=[str(ROOT / "gafime" / "_dummy.c")])],
+    cmdclass={"build_ext": RocmPayloadBuildExt},
+)
+
+if sys.platform == "linux":
+    setup_kwargs["options"] = {
+        "bdist_wheel": {
+            "plat_name": os.environ.get("GAFIME_ROCM_PLAT_NAME", "manylinux_2_28_x86_64")
+        }
+    }
+
+setup(**setup_kwargs)