Skip to content

Commit 5097bcf

Browse files
more minor fixes
1 parent d65f80d commit 5097bcf

File tree

2 files changed

+23
-24
lines changed

2 files changed

+23
-24
lines changed

numba_cuda/numba/cuda/tests/cudadrv/test_nvjitlink.py

Lines changed: 22 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -10,27 +10,28 @@
1010
from numba import config
1111

1212
TEST_BIN_DIR = os.getenv("NUMBA_CUDA_TEST_BIN_DIR")
13-
test_device_functions_a = os.path.join(
14-
TEST_BIN_DIR, "test_device_functions.a"
15-
)
16-
test_device_functions_cubin = os.path.join(
17-
TEST_BIN_DIR, "test_device_functions.cubin"
18-
)
19-
test_device_functions_cu = os.path.join(
20-
TEST_BIN_DIR, "test_device_functions.cu"
21-
)
22-
test_device_functions_fatbin = os.path.join(
23-
TEST_BIN_DIR, "test_device_functions.fatbin"
24-
)
25-
test_device_functions_o = os.path.join(
26-
TEST_BIN_DIR, "test_device_functions.o"
27-
)
28-
test_device_functions_ptx = os.path.join(
29-
TEST_BIN_DIR, "test_device_functions.ptx"
30-
)
31-
test_device_functions_ltoir = os.path.join(
32-
TEST_BIN_DIR, "test_device_functions.ltoir"
33-
)
13+
if TEST_BIN_DIR:
14+
test_device_functions_a = os.path.join(
15+
TEST_BIN_DIR, "test_device_functions.a"
16+
)
17+
test_device_functions_cubin = os.path.join(
18+
TEST_BIN_DIR, "test_device_functions.cubin"
19+
)
20+
test_device_functions_cu = os.path.join(
21+
TEST_BIN_DIR, "test_device_functions.cu"
22+
)
23+
test_device_functions_fatbin = os.path.join(
24+
TEST_BIN_DIR, "test_device_functions.fatbin"
25+
)
26+
test_device_functions_o = os.path.join(
27+
TEST_BIN_DIR, "test_device_functions.o"
28+
)
29+
test_device_functions_ptx = os.path.join(
30+
TEST_BIN_DIR, "test_device_functions.ptx"
31+
)
32+
test_device_functions_ltoir = os.path.join(
33+
TEST_BIN_DIR, "test_device_functions.ltoir"
34+
)
3435

3536

3637
@unittest.skipIf(

numba_cuda/numba/cuda/tests/test_binary_generation/Makefile

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,4 @@ all:
4949
nvcc $(NVCC_FLAGS) $(LTOIR_FLAGS) -o $(OUTPUT_DIR)/test_device_functions.ltoir.o test_device_functions.cu
5050
# Generate LTO-IR in a "raw" LTO-IR container
5151
python generate_raw_ltoir.py --arch sm_$(GPU_CC) -o $(OUTPUT_DIR)/test_device_functions.ltoir test_device_functions.cu
52-
# We also want to test linking a .cu file; this needs no compilation,
53-
# so copy it instead
54-
cp test_device_functions.cu $(OUTPUT_DIR)
52+

0 commit comments

Comments
 (0)