Function _skip_for_invalid_environment

tests/unit/ops/aio/test_aio.py:26–29 · view source on GitHub ↗

(use_cuda_pinned_tensor=True)

Source from the content-addressed store, hash-verified

24
25
26	def _skip_for_invalid_environment(use_cuda_pinned_tensor=True):
27	if get_accelerator().device_name() != 'cuda':
28	if use_cuda_pinned_tensor:
29	pytest.skip("torch.pin_memory is only supported in CUDA environments.")
30
31
32	def _get_local_rank():

test_parallel_readMethod · 0.85

test_async_readMethod · 0.85

test_parallel_writeMethod · 0.85

test_async_writeMethod · 0.85

test_readMethod · 0.85

test_writeMethod · 0.85

test_offset_writeMethod · 0.85

test_offset_readMethod · 0.85

get_acceleratorFunction · 0.90

device_nameMethod · 0.45

no test coverage detected

searching dependent graphs…