Home
last modified time | relevance | path

Searched defs:MM0 (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
H A Dkernel_forward.h347 struct MM0 { struct
355 using GemmType = DefaultGemmType<ArchTag, scalar_t>;
357 using OpClass = typename GemmType::OpClass;
358 using DefaultConfig =
367 static constexpr int kAlignmentA =
369 static constexpr int kAlignmentB =
371 using ThreadblockShape = cutlass::gemm::
373 using WarpShape = cutlass::gemm::GemmShape<32, 32, GemmType::WarpK>;
374 using DefaultMma = typename cutlass::gemm::threadblock::FindDefaultMma<
393 using MmaCore = typename DefaultMma::MmaCore;
[all …]
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/
H A DInstructionSimplify.cpp6016 auto *MM0 = dyn_cast<IntrinsicInst>(Op0); in foldMinMaxSharedOp() local
/aosp_15_r20/external/mesa3d/src/mesa/x86/
H A Dassyntax.h215 #define MM0 %mm0 macro
/aosp_15_r20/prebuilts/bazel/common/android_tools/android_tools/
Dall_android_tools_deploy.jarMETA-INF/ META-INF/MANIFEST.MF build-data.properties com/ com ...