Searched defs:cuda_init_mode (Results 1 – 1 of 1) sorted by relevance
168 self, cuda_init_mode=CUDAInitMode.CUDA_BEFORE, nested=False argument200 def test_error_already_wrapped(self, nested, cuda_init_mode): argument708 def test_auto_wrap_smoke_test(self, cuda_init_mode, cpu_offload, use_device_id): argument