Home
last modified time | relevance | path

Searched refs:deterministicAlgorithmsWarnOnly (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DContext.cpp63 bool Context::deterministicAlgorithmsWarnOnly() const { in deterministicAlgorithmsWarnOnly() function in at::Context
82 if (globalContext().deterministicAlgorithmsWarnOnly()) { in alertNotDeterministic()
184 if (deterministicAlgorithmsWarnOnly()) { in alertCuBLASConfigNotDeterministic()
H A DContext.h253 bool deterministicAlgorithmsWarnOnly() const;
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/
H A Dattention_backward.cu101 if (ctx.deterministicAlgorithmsWarnOnly()) { in _flash_attention_backward()
187 if (ctx.deterministicAlgorithmsWarnOnly()) { in _scaled_dot_product_cudnn_attention_backward_cuda()
654 if (ctx.deterministicAlgorithmsWarnOnly()) { in _efficient_attention_backward()
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dguards.cpp519 _deterministic_algorithms_warn_only = ctx.deterministicAlgorithmsWarnOnly(); in init()
535 ctx.deterministicAlgorithmsWarnOnly() && in check()
553 ctx.deterministicAlgorithmsWarnOnly()) in reason()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp902 if (at::globalContext().deterministicAlgorithmsWarnOnly()) { in THPModule_deterministicAlgorithmsWarnOnly()