Home
last modified time | relevance | path

Searched refs:MMALayout (Results 1 – 7 of 7) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/
H A DWmmaOpsToNvvm.cpp104 NVVM::MMALayout layout = NVVM::MMALayout::row; in matchAndRewrite()
152 NVVM::MMALayout layout = NVVM::MMALayout::row; in matchAndRewrite()
223 NVVM::MMALayout layout = NVVM::MMALayout::row; in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Target/LLVMIR/Dialect/NVVM/
H A DNVVMToLLVMIRTranslation.cpp68 static llvm::Intrinsic::ID getLdMatrixIntrinsicId(NVVM::MMALayout layout, in getLdMatrixIntrinsicId()
70 if (layout == NVVM::MMALayout::row) { in getLdMatrixIntrinsicId()
/llvm-project-15.0.7/mlir/lib/Conversion/NVGPUToNVVM/
H A DNVGPUToNVVM.cpp231 /*layout=*/op.getTranspose() ? NVVM::MMALayout::col in matchAndRewrite()
232 : NVVM::MMALayout::row); in matchAndRewrite()
323 std::array<NVVM::MMALayout, 2>{NVVM::MMALayout::row, in matchAndRewrite()
324 NVVM::MMALayout::col}); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DNvGpuSupport.h75 NVVM::MMALayout targetLayout;
H A DNvGpuSupport.cpp215 params.targetLayout = NVVM::MMALayout::row; in getLdMatrixParams()
217 params.targetLayout = NVVM::MMALayout::col; in getLdMatrixParams()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/LLVMIR/
H A DNVVMOps.td594 def MMALayout : I32EnumAttr<"MMALayout", "NVVM MMA layout",
599 def MMALayoutAttr : EnumAttr<NVVM_Dialect, MMALayout, "mma_layout"> {
659 "int m, int n, int k, mlir::NVVM::MMALayout layoutEnum,"
729 "int m, int n, int k, mlir::NVVM::MMALayout layoutEnum,"
786 "int m, int n, int k, mlir::NVVM::MMALayout layoutAEnum,"
787 "mlir::NVVM::MMALayout layoutBEnum, mlir::NVVM::MMATypes eltypeAEnum,"
962 mlir::NVVM::MMALayout layoutAEnum, mlir::NVVM::MMALayout layoutBEnum,
989 "Optional<std::array<MMALayout, 2>>":$multiplicandLayouts)>
/llvm-project-15.0.7/mlir/lib/Dialect/LLVMIR/IR/
H A DNVVMDialect.cpp194 Optional<std::array<MMALayout, 2>> multiplicandLayouts) { in build() argument
223 result.addAttribute("layoutA", MMALayoutAttr::get(ctx, MMALayout::row)); in build()
224 result.addAttribute("layoutB", MMALayoutAttr::get(ctx, MMALayout::col)); in build()