1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
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",
],
)
|
