Home
last modified time | relevance | path

Searched refs:deterministicAlgorithms (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DContext.cpp59 bool Context::deterministicAlgorithms() const { in deterministicAlgorithms() function in at::Context
81 if (globalContext().deterministicAlgorithms()) { in alertNotDeterministic()
170 if (C10_LIKELY(!deterministicAlgorithms() || cublas_config_deterministic)) { in alertCuBLASConfigNotDeterministic()
H A DContext.h252 bool deterministicAlgorithms() const;
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DEmptyTensor.cpp68 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in empty_mps()
111 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in empty_strided_mps()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/
H A DDeconv.cpp187 if(at::globalContext().deterministicAlgorithms() || at::globalContext().deterministicMkldnn()) in deconvolution()
269 if(at::globalContext().deterministicAlgorithms() || at::globalContext().deterministicMkldnn()) in deconvolution_backward_data()
365 if(at::globalContext().deterministicAlgorithms() || at::globalContext().deterministicMkldnn()) in deconvolution_backward_weights()
H A DConv.cpp199 if(at::globalContext().deterministicAlgorithms() || at::globalContext().deterministicMkldnn()){ in convolution()
286 if(at::globalContext().deterministicAlgorithms() || at::globalContext().deterministicMkldnn()){ in convolution_backward_weights()
383 if(at::globalContext().deterministicAlgorithms() || at::globalContext().deterministicMkldnn()){ in convolution_backward_data()
H A DMatmul.cpp180 if(at::globalContext().deterministicAlgorithms() || at::globalContext().deterministicMkldnn()) in matmul()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorFactories.cu57 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in empty_cuda()
83 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in empty_strided_cuda()
H A DResize.cpp69 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in resize_cuda_()
H A DIndexing.cu863 if (globalContext().deterministicAlgorithms()){ in index_add_cuda_impl()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DIndexKernel.cpp118 bool is_deterministic = at::globalContext().deterministicAlgorithms(); in put_kernel()
165 const bool is_deterministic = at::globalContext().deterministicAlgorithms(); in index_put_kernel()
308 bool is_deterministic = at::globalContext().deterministicAlgorithms(); in index_copy_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/
H A Dattention_backward.cu100 if (ctx.deterministicAlgorithms()) { in _flash_attention_backward()
186 if (ctx.deterministicAlgorithms()) { in _scaled_dot_product_cudnn_attention_backward_cuda()
653 if (ctx.deterministicAlgorithms()) { in _efficient_attention_backward()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCublasHandlePool.cpp184 if (at::globalContext().deterministicAlgorithms()) { in getCurrentCUDABlasHandle()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DTensorFactory.cpp125 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in resize_mps_()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DTensorAdvancedIndexing.cpp193 if (self.device().type() == DeviceType::CUDA && globalContext().deterministicAlgorithms()) { in _index_put_impl_quantized_cuda_()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DResize.cpp253 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in _resize_()
H A DTensorAdvancedIndexing.cpp814 …device().type() == DeviceType::XPU) && (accumulate || globalContext().deterministicAlgorithms())) { in _index_put_impl_()
876 if (result.is_cuda() && globalContext().deterministicAlgorithms()){ in TORCH_IMPL_FUNC()
1811 …bool deterministic = globalContext().deterministicAlgorithms() && (self.device().type() == DeviceT… in scatter_impl()
1905 if (globalContext().deterministicAlgorithms() && self.device().type() == DeviceType::CUDA) { in TORCH_IMPL_FUNC()
H A DTensorFactories.cpp251 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in empty_cpu()
323 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in empty_strided_cpu()
344 …if (C10_UNLIKELY(at::globalContext().deterministicAlgorithms() && at::globalContext().deterministi… in empty_out()
H A DConvolution.cpp1183 bool deterministic = ctx.deterministicCuDNN() || ctx.deterministicAlgorithms(); in convolution()
1329 params.deterministic = ctx.deterministicCuDNN() || ctx.deterministicAlgorithms(); in select_conv_backend()
2015 params.deterministic = ctx.deterministicCuDNN() || ctx.deterministicAlgorithms(); in convolution_backward()
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dguards.cpp518 _deterministic_algorithms = ctx.deterministicAlgorithms(); in init()
533 _deterministic_algorithms == ctx.deterministicAlgorithms() && in check()
550 if (_deterministic_algorithms != ctx.deterministicAlgorithms()) in reason()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCUDATensorMath.cu839 bool deterministic = globalContext().deterministicAlgorithms(); in bmm_out_sparse_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DIndexing.mm85 …const bool serial_index_put = at::globalContext().deterministicAlgorithms() && !accumulate && !ind…
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp893 if (at::globalContext().deterministicAlgorithms()) { in THPModule_deterministicAlgorithms()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp4226 const bool is_deterministic = at::globalContext().deterministicAlgorithms();