41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
dsa_tests = [
|
|
"impl/CUDAAssertionsTest_1_var_test.cu",
|
|
"impl/CUDAAssertionsTest_catches_stream.cu",
|
|
"impl/CUDAAssertionsTest_catches_thread_and_block_and_device.cu",
|
|
"impl/CUDAAssertionsTest_from_2_processes.cu",
|
|
"impl/CUDAAssertionsTest_multiple_writes_from_blocks_and_threads.cu",
|
|
"impl/CUDAAssertionsTest_multiple_writes_from_multiple_blocks.cu",
|
|
"impl/CUDAAssertionsTest_multiple_writes_from_same_block.cu",
|
|
]
|
|
|
|
def define_targets(rules, gtest_deps):
|
|
rules.cc_test(
|
|
name = "test",
|
|
srcs = [
|
|
"impl/CUDATest.cpp",
|
|
],
|
|
target_compatible_with = rules.requires_cuda_enabled(),
|
|
deps = [
|
|
"//c10/cuda",
|
|
] + gtest_deps,
|
|
)
|
|
|
|
for src in dsa_tests:
|
|
name = src.replace("impl/", "").replace(".cu", "")
|
|
rules.cuda_library(
|
|
name = "test_" + name + "_lib",
|
|
srcs = [
|
|
src,
|
|
],
|
|
target_compatible_with = rules.requires_cuda_enabled(),
|
|
deps = [
|
|
"//c10/cuda",
|
|
] + gtest_deps,
|
|
)
|
|
rules.cc_test(
|
|
name = "test_" + name,
|
|
deps = [
|
|
":test_" + name + "_lib",
|
|
],
|
|
)
|