Home
last modified time | relevance | path

Searched defs:MM1 (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.h430 struct MM1 { struct
435 using GemmType = DefaultGemmType<ArchTag, scalar_t>;
437 using OpClass = typename GemmType::OpClass;
438 using DefaultConfig =
447 static constexpr int kAlignmentA = DefaultConfig::kAlignmentA; // from smem
448 static constexpr int kAlignmentB =
450 using ThreadblockShape = cutlass::gemm::
452 using WarpShape = cutlass::gemm::GemmShape<32, 32, GemmType::WarpK>;
453 using InstructionShape = typename GemmType::InstructionShape;
455 using LayoutB = cutlass::layout::RowMajor;
[all …]
/aosp_15_r20/external/autotest/client/cros/cellular/
H A Dmm1_constants.py13 MM1 = '/org/freedesktop/ModemManager1' variable
/aosp_15_r20/external/python/cpython3/Lib/test/
Dtest_typing.py3567 class MM1(MutableMapping[str, str], collections.abc.MutableMapping): class
/aosp_15_r20/external/mesa3d/src/mesa/x86/
H A Dassyntax.h216 #define MM1 %mm1 macro