xref: /aosp_15_r20/external/pytorch/c10/cuda/test/build.bzl (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

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",
            ],
        )