Home
last modified time | relevance | path

Searched defs:device_guard (Results 1 – 25 of 47) sorted by relevance

12

/external/pytorch/aten/src/ATen/test/
Dcuda_stream_test.cpp136 at::cuda::CUDAGuard device_guard(1); in TEST() local
207 at::cuda::CUDAGuard device_guard{1}; in TEST() local
303 at::cuda::CUDAGuard device_guard(0); in TEST() local
330 at::cuda::CUDAGuard device_guard(0); in TEST() local
334 at::cuda::CUDAGuard device_guard(1); in TEST() local
356 at::cuda::CUDAGuard device_guard(0); in TEST() local
398 at::cuda::CUDAGuard device_guard(0); in TEST() local
414 at::cuda::CUDAGuard device_guard(0); in TEST() local
422 at::cuda::CUDAGuard device_guard(0); in TEST() local
/external/pytorch/aten/src/ATen/
DTensorIndexing.cpp69 OptionalDeviceGuard device_guard(device_of(*this)); in index() local
78 OptionalDeviceGuard device_guard(device_of(*this)); in index_put_() local
84 OptionalDeviceGuard device_guard(device_of(*this)); in index_put_() local
/external/pytorch/c10/xpu/test/impl/
DXPUGuardTest.cpp19 const c10::DeviceGuard device_guard(device); in TEST() local
38 const c10::DeviceGuard device_guard(device); in TEST() local
/external/pytorch/torch/csrc/autograd/
Dinput_buffer.cpp199 c10::OptionalDeviceGuard device_guard{opt_sync_stream->device()}; in add() local
219 c10::OptionalDeviceGuard device_guard{device_of(var)}; in add() local
Dpython_torch_functions_manual.cpp55 OptionalDeviceGuard device_guard(device_of(result)); in dispatch_range() local
66 DeviceGuard device_guard(options.device()); in dispatch_range() local
152 OptionalDeviceGuard device_guard(device_of(self)); in dispatch_nonzero() local
158 OptionalDeviceGuard device_guard(device_of(self)); in dispatch_nonzero() local
164 OptionalDeviceGuard device_guard(device_of(self)); in dispatch_nonzero_numpy() local
Dpython_variable_indexing.cpp342 OptionalDeviceGuard device_guard(device_of(self_)); in THPVariable_getitem() local
448 OptionalDeviceGuard device_guard(device_of(self_)); in THPVariable_setitem() local
/external/pytorch/aten/src/ATen/cuda/
DEmptyTensor.cpp16 const DeviceGuard device_guard(device); in empty_cuda() local
56 const DeviceGuard device_guard(device); in empty_strided_cuda() local
DCUDAGraph.cpp227 c10::OptionalDeviceGuard device_guard{capture_stream_.device()}; in replay() local
/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
DOpenRegMem.cpp73 const c10::DeviceGuard device_guard(device); in empty_openreg() local
91 const c10::DeviceGuard device_guard(device); in empty_strided_openreg() local
/external/pytorch/torch/_inductor/codegen/cuda/
Ddevice_op_overrides.py15 def device_guard(self, device_idx): member in CUDADeviceOpOverrides
/external/pytorch/torch/_inductor/codegen/xpu/
Ddevice_op_overrides.py15 def device_guard(self, device_idx): member in XPUDeviceOpOverrides
/external/pytorch/torch/csrc/distributed/c10d/quantization/
Dquantization_gpu.cu60 at::cuda::OptionalCUDAGuard device_guard; in _float_to_bfloat16_cuda() local
111 at::cuda::OptionalCUDAGuard device_guard; in _bfloat16_to_float_cuda() local
/external/pytorch/aten/src/ATen/native/cuda/
DCopy.cu188 CUDAGuard device_guard(src_device); in copy_device_to_device() local
345 cuda::OptionalCUDAGuard device_guard; in copy_kernel_cuda() local
DJitLoops.cuh174 const OptionalDeviceGuard device_guard(iter.device(1)); in opmath_jitted_gpu_kernel_with_scalars() local
DCompareKernels.cu63 const DeviceGuard device_guard(iter.device(1)); in compare_kernel_impl() local
DLoops.cuh185 const OptionalDeviceGuard device_guard(iter.device(1)); in opmath_gpu_kernel_with_scalars() local
210 OptionalDeviceGuard device_guard; in opmath_symmetric_gpu_kernel_with_scalars() local
DAmpKernels.cu50 const OptionalDeviceGuard device_guard(device_of(scaled_grad)); in _amp_non_finite_check_and_unscale_cuda_() local
/external/pytorch/torch/csrc/utils/
Dtensor_new.cpp647 at::OptionalDeviceGuard device_guard(deviceOptional); in legacy_sparse_tensor_generic_ctor_new() local
659 at::OptionalDeviceGuard device_guard(deviceOptional); in legacy_sparse_tensor_generic_ctor_new() local
740 at::OptionalDeviceGuard device_guard(deviceOptional); in legacy_tensor_generic_ctor_new() local
955 at::OptionalDeviceGuard device_guard(r.deviceOptional(ARG_DEVICE)); in sparse_compressed_tensor_ctor_worker() local
1010 at::OptionalDeviceGuard device_guard(r.deviceOptional(ARG_DEVICE1)); in sparse_compressed_tensor_ctor_worker() local
1182 at::OptionalDeviceGuard device_guard(r.deviceOptional(ARG_DEVICE)); in sparse_coo_tensor_ctor() local
1216 at::OptionalDeviceGuard device_guard(r.deviceOptional(ARG_DEVICE1)); in sparse_coo_tensor_ctor() local
1249 at::OptionalDeviceGuard device_guard(r.deviceOptional(ARG_DEVICE2)); in sparse_coo_tensor_ctor() local
/external/pytorch/torch/csrc/cuda/
Dnccl.cpp588 at::cuda::OptionalCUDAGuard device_guard; in broadcast() local
640 at::cuda::OptionalCUDAGuard device_guard; in reduce() local
694 at::cuda::OptionalCUDAGuard device_guard; in all_reduce() local
736 at::cuda::OptionalCUDAGuard device_guard; in reduce_scatter() local
777 at::cuda::OptionalCUDAGuard device_guard; in all_gather() local
/external/pytorch/tools/autograd/templates/
Dpython_variable_methods.cpp250 OptionalDeviceGuard device_guard(device_of(self)); in dispatch_contiguous() local
294 OptionalDeviceGuard device_guard(device_of(self)); in dispatch_copy_() local
320 OptionalDeviceGuard device_guard(device_of(self)); in dispatch_to() local
383 OptionalDeviceGuard device_guard(device_of(self)); in dispatch_invert() local
448 OptionalDeviceGuard device_guard(device_of(self)); in dispatch_nonzero() local
454 OptionalDeviceGuard device_guard(device_of(self)); in dispatch_nonzero_numpy() local
953 OptionalDeviceGuard device_guard(device_of(self_)); in THPVariable_new() local
965 OptionalDeviceGuard device_guard(device_of(self_)); in THPVariable_new_tensor() local
/external/pytorch/aten/src/ATen/native/quantized/cuda/
DFusedObsFakeQuant.cu140 at::cuda::OptionalCUDAGuard device_guard; in _calculate_moving_average() local
196 at::cuda::OptionalCUDAGuard device_guard; in _calc_moving_avg_qparams_helper() local
/external/pytorch/aten/src/ATen/cuda/detail/
DCUDAHooks.cpp132 at::OptionalDeviceGuard device_guard; in isPinnedPtr() local
458 at::DeviceGuard device_guard(at::Device(at::DeviceType::CUDA, device_index)); in deviceSynchronize() local
/external/pytorch/aten/src/ATen/mps/
DEmptyTensor.cpp105 const DeviceGuard device_guard(device); in empty_strided_mps() local
/external/pytorch/test/inductor/extension_backends/triton/
Dextension_codegen_backend.py48 def device_guard(self, device_idx: int) -> str: # noqa: ARG002 unused-argument member in CPUDeviceOpOverrides
/external/pytorch/torch/csrc/profiler/stubs/
Dcuda.cpp89 at::cuda::OptionalCUDAGuard device_guard; in onEachDevice() local

12