We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d172e3b commit 081d63cCopy full SHA for 081d63c
1 file changed
.github/workflows/verify_extension_build.yml
@@ -31,6 +31,24 @@ jobs:
31
pip install -r .github/workflows/requirements_cuda_ci.txt
32
pip install -e .
33
34
+ - name: Validate CUDA environment
35
+ run: |
36
+ python - <<'PY'
37
+ import os
38
+ import sys
39
+ import torch
40
+
41
+ cuda_home = os.environ.get("CUDA_HOME")
42
+ if not cuda_home:
43
+ sys.exit("CUDA_HOME is not set; CUDA toolkit may not be available.")
44
+ print(f"CUDA_HOME={cuda_home}")
45
46
+ cuda_version = getattr(torch.version, "cuda", None)
47
+ if not cuda_version:
48
+ sys.exit("torch.version.cuda is empty; torch may not have CUDA support.")
49
+ print(f"torch.version.cuda={cuda_version}")
50
+ PY
51
52
- name: Test extension build via import
53
run: |
- pytest tests/import_test.py -k test_extension_built
54
+ pytest tests/import_test.py -k test_extension_built
0 commit comments