| |
| """Verify that EDGS CUDA extension wheels are installed and importable.""" |
|
|
| from __future__ import annotations |
|
|
| import importlib |
| import sys |
|
|
|
|
| def check_module(name: str) -> bool: |
| try: |
| module = importlib.import_module(name) |
| print(f"OK: {name} -> {getattr(module, '__file__', '<builtin>')}") |
| return True |
| except Exception as error: |
| print(f"FAIL: {name}: {error!r}") |
| return False |
|
|
|
|
| def main() -> int: |
| names = [ |
| "diff_gaussian_rasterization", |
| "diff_gaussian_rasterization._C", |
| "simple_knn._C", |
| ] |
| ok = all(check_module(name) for name in names) |
| if not ok: |
| print("\nInstall compatible prebuilt wheels from GitHub Releases and rerun this script.") |
| return 1 |
| print("\nEDGS CUDA extension imports passed.") |
| return 0 |
|
|
|
|
| if __name__ == "__main__": |
| raise SystemExit(main()) |
|
|
