Home
last modified time | relevance | path

Searched defs:use_fast_accum (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DRowwiseScaledMM.cu461 bool use_fast_accum, in dispatch_fp8_rowwise_kernel_on_fast_accum()
481 bool use_fast_accum, in dispatch_fp8_rowwise_kernel_on_input_dtypes()
502 bool use_fast_accum, in dispatch_fp8_rowwise_kernel_on_bias_dtype()
575 bool use_fast_accum, in f8f8bf16_rowwise()
H A DBlas.cpp993 bool use_fast_accum, in _scaled_mm_out_cuda()
1196 bool use_fast_accum) { in _scaled_mm_cuda()
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dshim_common.cpp691 int8_t use_fast_accum, in aoti_torch__scaled_mm()
723 int8_t use_fast_accum, in aoti_torch__scaled_mm_v2()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/
H A DGemmCommon.h326 bool use_fast_accum; member
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDABlas.cpp1412 bool use_fast_accum) { in scaled_gemm()