Home
last modified time | relevance | path

Searched defs:TEST_CUDA (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_cpp_extensions_stream_and_event.py26 TEST_CUDA = TEST_CUDA and CUDA_HOME is not None variable
H A Dtest_cuda_nvml_based_avail.py32 TEST_CUDA = torch.cuda.is_available() variable
H A Dtest_cpp_extensions_mtia_backend.py25 TEST_CUDA = TEST_CUDA and CUDA_HOME is not None variable
H A Dtest_cpp_extensions_open_device_registration.py29 TEST_CUDA = TEST_CUDA and CUDA_HOME is not None variable
H A Dtest_cpp_extensions_jit.py32 TEST_CUDA = TEST_CUDA and CUDA_HOME is not None variable
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py1346 TEST_CUDA = torch.cuda.is_available() variable