Home
last modified time | relevance | path

Searched defs:requires_cuda (Results 1 – 17 of 17) sorted by relevance

/external/pytorch/test/distributed/_composable/fsdp/
Dtest_fully_shard_logging.py13 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
/external/pytorch/test/inductor/
Dtest_minifier_isolate.py16 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_group_batch_fusion.py23 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_inductor_freezing.py38 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_cudagraph_trees.py53 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
/external/pytorch/test/dynamo/
Dtest_debug_utils.py13 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_minifier.py9 requires_cuda = unittest.skipUnless(torch.cuda.is_available(), "requires cuda") variable
Dtest_structured_trace.py28 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_backends.py18 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_logging.py28 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_activation_checkpointing.py33 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_subclasses.py144 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_higher_order_ops.py37 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dtest_repros.py61 requires_cuda = unittest.skipUnless(torch.cuda.is_available(), "requires cuda") variable
/external/pytorch/torch/testing/_internal/
Dtriton_utils.py9 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
Dcommon_utils.py1458 requires_cuda = unittest.skipUnless(torch.cuda.is_available(), "Requires CUDA") variable
/external/pytorch/test/export/
Dtest_converter.py20 requires_cuda = unittest.skipUnless(torch.cuda.is_available(), "requires cuda") variable