Skip to content

Commit 7fa3f76

Browse files
committed
Remove pytest_skipif_cuda_include_not_found() and get_cuda_home() entirely under cuda_bindings/examples/
1 parent 95024c4 commit 7fa3f76

2 files changed

Lines changed: 0 additions & 20 deletions

File tree

cuda_bindings/examples/3_CUDA_Features/globalToShmemAsyncCopy_test.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1114,7 +1114,6 @@ def MatrixMultiply(dimsA, dimsB, kernel_number):
11141114

11151115

11161116
def main():
1117-
common.pytest_skipif_cuda_include_not_found()
11181117
common.pytest_skipif_compute_capability_too_low(findCudaDevice(), (7, 0))
11191118

11201119
print("[globalToShmemAsyncCopy] - Starting...")

cuda_bindings/examples/common/common.py

Lines changed: 0 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
# Copyright 2021-2025 NVIDIA Corporation. All rights reserved.
22
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
33

4-
import os
54

65
import numpy as np
76
from common.helper_cuda import checkCudaErrors
@@ -11,24 +10,6 @@
1110
from cuda.bindings import runtime as cudart
1211

1312

14-
def get_cuda_home():
15-
cuda_home = os.getenv("CUDA_HOME")
16-
if cuda_home is None:
17-
cuda_home = os.getenv("CUDA_PATH")
18-
return cuda_home
19-
20-
21-
def pytest_skipif_cuda_include_not_found():
22-
import pytest
23-
24-
cuda_home = get_cuda_home()
25-
if cuda_home is None:
26-
pytest.skip("CUDA_HOME/CUDA_PATH not set")
27-
cuda_include = os.path.join(cuda_home, "include")
28-
if not os.path.exists(cuda_include):
29-
pytest.skip(f"$CUDA_HOME/include does not exist: '{cuda_include}'")
30-
31-
3213
def pytest_skipif_compute_capability_too_low(devID, required_cc_major_minor):
3314
import pytest
3415

0 commit comments

Comments
 (0)