Home
last modified time | relevance | path

Searched refs:eltype (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/
H A DWmmaOpsToNvvm.cpp88 NVVM::MMATypes eltype = getElementType(retType); in matchAndRewrite() local
94 n = NVVM::WMMALoadOp::inferNDimension(m, k, eltype); in matchAndRewrite()
98 m = NVVM::WMMALoadOp::inferMDimension(k, n, eltype); in matchAndRewrite()
102 k = NVVM::WMMALoadOp::inferKDimension(m, n, eltype); in matchAndRewrite()
107 if (NVVM::WMMALoadOp::getIntrinsicID(m, n, k, layout, eltype, frag) == 0) in matchAndRewrite()
122 op, resType, dataPtr, leadingDim, m, n, k, layout, eltype, frag); in matchAndRewrite()
153 NVVM::MMATypes eltype = getElementType(srcType); in matchAndRewrite() local
156 int64_t k = NVVM::WMMAStoreOp::inferKDimension(m, n, eltype); in matchAndRewrite()
157 if (NVVM::WMMAStoreOp::getIntrinsicID(m, n, k, layout, eltype) == 0) in matchAndRewrite()
175 op, dataPtr, m, n, k, layout, eltype, storeOpOperands, leadingDim); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/LLVMIR/
H A DNVVMOps.td378 frag.ptx_elt_type # "\" == eltype)"
390 frag.ptx_elt_type # "\" == eltype)"
400 frag.ptx_elt_type # "\" == eltype)"
410 frag.ptx_elt_type # "\" == eltype)"
646 MMATypesAttr:$eltype, MMAFragAttr:$frag)> {
662 "llvm::StringRef eltype = stringifyEnum(eltypeEnum);"
669 " llvm::StringRef eltype = stringifyEnum(eltypeEnum);"
675 " llvm::StringRef eltype = stringifyEnum(eltypeEnum);"
681 " llvm::StringRef eltype = stringifyEnum(eltypeEnum);"
690 $m, $n, $k, $layout, $eltype, $frag);
[all …]
/llvm-project-15.0.7/mlir/test/Conversion/GPUToNVVM/
H A Dwmma-ops-to-nvvm.mlir23 …// CHECK-SAME: {eltype = #nvvm.mma_type<f16>, frag = #nvvm.mma_frag<a>, k = 16 : i32, layout = #nv…
35 …// CHECK32-SAME: {eltype = #nvvm.mma_type<f16>, frag = #nvvm.mma_frag<a>, k = 16 : i32, layout = #…
67 …// CHECK-SAME: {eltype = #nvvm.mma_type<f16>, k = 16 : i32, layout = #nvvm.mma_layout<row>, m = 16…
83 …// CHECK32-SAME: {eltype = #nvvm.mma_type<f16>, k = 16 : i32, layout = #nvvm.mma_layout<row>, m = …
130 // CHECK: %[[C:.+]] = nvvm.wmma.load %{{.*}}, %{{.*}} {eltype = #nvvm.mma_type<f16>, frag =…
135 // CHECK: %[[A:.+]] = nvvm.wmma.load %{{.*}}, %{{.*}} {eltype = #nvvm.mma_type<f16>, frag =…
136 // CHECK: %[[B:.+]] = nvvm.wmma.load %{{.*}}, %{{.*}} {eltype = #nvvm.mma_type<f16>, frag =…
164 // CHECK: nvvm.wmma.store %{{.*}}, %{{.*}}, %[[E0]], %[[E1]], %[[E2]], %[[E3]] {eltype = #n…
/llvm-project-15.0.7/mlir/test/Target/LLVMIR/
H A Dnvvmir.mlir245 …{eltype = #nvvm.mma_type<f16>, frag = #nvvm.mma_frag<a>, k = 16 : i32, layout = #nvvm.mma_layout<r…
259 …{eltype = #nvvm.mma_type<f16>, k = 16 : i32, layout = #nvvm.mma_layout<row>, m = 16 : i32, n = 16 …
292 …{eltype = #nvvm.mma_type<tf32>, frag = #nvvm.mma_frag<a>, k = 8 : i32, layout = #nvvm.mma_layout<r…
/llvm-project-15.0.7/mlir/test/Dialect/LLVMIR/
H A Dinvalid.mlir1037 …{eltype = #nvvm.mma_type<f16>, frag = #nvvm.mma_frag<a>, k = 16 : i32, layout = #nvvm.mma_layout<r…
1047 …{eltype = #nvvm.mma_type<f16>, frag = #nvvm.mma_frag<a>, k = 16 : i32, layout = #nvvm.mma_layout<r…
1057 …{eltype = #nvvm.mma_type<f16>, frag = #nvvm.mma_frag<b>, k = 16 : i32, layout = #nvvm.mma_layout<r…
1068 …{eltype = #nvvm.mma_type<f16>, frag = #nvvm.mma_frag<c>, k = 16 : i32, layout = #nvvm.mma_layout<r…
1081 …{eltype = #nvvm.mma_type<f16>, k = 16 : i32, layout = #nvvm.mma_layout<row>, m = 16 : i32, n = 16 …
H A Dnvvm.mlir263 …// CHECK: nvvm.wmma.load {{.*}} {eltype = #nvvm.mma_type<tf32>, frag = #nvvm.mma_frag<a>, k = 8 : …
265 …{eltype = #nvvm.mma_type<tf32>, frag = #nvvm.mma_frag<a>, k = 8 : i32, layout = #nvvm.mma_layout<r…