gafime-cuda 0.4.7__tar.gz

gafime_cuda-0.4.7/MANIFEST.in

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

gafime_cuda-0.4.7/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.4
+Name: gafime-cuda
+Version: 0.4.7
+Summary: NVIDIA CUDA 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-cuda
+
+Vendor GPU runtime payload for GAFIME 0.4.7.
+
+This package is generated from the GAFIME source tree during CI and carries
+only the CUDA native runtime payload. Install the base package
+with `gafime`; use this package only for the matching GPU runtime.

gafime_cuda-0.4.7/README.md

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

gafime_cuda-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_cuda-0.4.7/gafime_cuda/__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_cuda.dll",
+        base / "libgafime_cuda.so",
+        base / "gafime_cuda.so",
+        base / "gafime_cuda.pyd",
+    ]

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

@@ -0,0 +1,20 @@
+Metadata-Version: 2.4
+Name: gafime-cuda
+Version: 0.4.7
+Summary: NVIDIA CUDA 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-cuda
+
+Vendor GPU runtime payload for GAFIME 0.4.7.
+
+This package is generated from the GAFIME source tree during CI and carries
+only the CUDA native runtime payload. Install the base package
+with `gafime`; use this package only for the matching GPU runtime.

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

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

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

@@ -0,0 +1 @@
+

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

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

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

@@ -0,0 +1 @@
+gafime_cuda

gafime_cuda-0.4.7/pyproject.toml

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

gafime_cuda-0.4.7/setup.cfg

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

gafime_cuda-0.4.7/setup.py

@@ -0,0 +1,105 @@
+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
+
+
+def _find_nvcc() -> str | None:
+    nvcc = shutil.which("nvcc")
+    if nvcc:
+        return nvcc
+    exe_name = "nvcc.exe" if sys.platform == "win32" else "nvcc"
+    for env_name in ("CUDA_PATH", "CUDA_HOME"):
+        cuda_root = os.environ.get(env_name)
+        if not cuda_root:
+            continue
+        candidate = Path(cuda_root) / "bin" / exe_name
+        if candidate.exists():
+            return str(candidate)
+    return None
+
+
+class CudaPayloadBuildExt(build_ext):
+    def run(self):
+        package_dir = Path(self.build_lib) / "gafime_cuda"
+        package_dir.mkdir(parents=True, exist_ok=True)
+        self.output_dir = package_dir
+        self.build_cuda_backend()
+        super().run()
+
+    def build_cuda_backend(self) -> None:
+        machine = platform.machine().lower()
+        if machine in {"aarch64", "arm64"} or machine.startswith("arm"):
+            raise RuntimeError(f"gafime-cuda does not support ARM target {platform.machine()}.")
+        nvcc = _find_nvcc()
+        if not nvcc:
+            raise RuntimeError(
+                "nvcc was not found. Install CUDA Toolkit 13.2+ to build gafime-cuda "
+                "or set CUDA_PATH/CUDA_HOME to the toolkit root."
+            )
+
+        src_dir = ROOT / "src"
+        cuda_source = src_dir / "cuda" / "kernels.cu"
+        if sys.platform == "win32":
+            output_file = self.output_dir / "gafime_cuda.dll"
+            compiler_flags = ["/MD", "/O2"]
+        else:
+            output_file = self.output_dir / "libgafime_cuda.so"
+            compiler_flags = ["-fPIC", "-O3"]
+
+        gencode_flags = [
+            "-gencode=arch=compute_75,code=sm_75",
+            "-gencode=arch=compute_80,code=sm_80",
+            "-gencode=arch=compute_86,code=sm_86",
+            "-gencode=arch=compute_89,code=sm_89",
+            "-gencode=arch=compute_90,code=sm_90",
+            "-gencode=arch=compute_100,code=sm_100",
+            "-gencode=arch=compute_120,code=sm_120",
+            "-gencode=arch=compute_120,code=compute_120",
+        ]
+        cmd = [
+            nvcc,
+            *gencode_flags,
+            "-O3",
+            "--shared",
+            "-DGAFIME_BUILDING_DLL",
+            "-cudart",
+            "static",
+            "-Xcompiler",
+            ",".join(compiler_flags),
+            "-I",
+            str(src_dir / "common"),
+            "-o",
+            str(output_file),
+            str(cuda_source),
+        ]
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        if result.returncode != 0:
+            raise RuntimeError(f"CUDA build failed\nSTDOUT:\n{result.stdout}\nSTDERR:\n{result.stderr}")
+
+
+setup(
+    name="gafime-cuda",
+    version=VERSION,
+    description="NVIDIA CUDA runtime payload for GAFIME",
+    long_description=(ROOT / "README.md").read_text(encoding="utf-8"),
+    long_description_content_type="text/markdown",
+    packages=["gafime_cuda"],
+    package_data={"gafime_cuda": ["*.so", "*.dll", "*.pyd"]},
+    include_package_data=False,
+    install_requires=[f"gafime=={VERSION}"],
+    python_requires=">=3.10",
+    ext_modules=[Extension("gafime_cuda._native", sources=[str(ROOT / "gafime" / "_dummy.c")])],
+    cmdclass={"build_ext": CudaPayloadBuildExt},
+)