Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/kernel/
H A DfpA_intB_gemm.h60 struct GemmFpAIntB { struct
62 using Mma = Mma_;
63 using Epilogue = Epilogue_;
64 using EpilogueOutputOp = typename Epilogue::OutputOp;
65 using ThreadblockSwizzle = ThreadblockSwizzle_;
66 static bool const kSplitKSerial = SplitKSerial;
68 using ElementA = typename Mma::IteratorA::Element;
69 using LayoutA = typename Mma::IteratorA::Layout;
70 using ElementB = typename Mma::IteratorB::Element;
71 using LayoutB = typename Mma::IteratorB::Element;
[all …]