Searched refs:aType (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/AMX/IR/ |
| H A D | AMXDialect.cpp | 74 VectorType aType = getLhsVectorType(); in verify() local
77 if (failed(verifyTileSize(*this, aType)) || in verify()
80 failed(verifyMultShape(*this, aType, bType, cType, 1))) in verify()
82 Type ta = aType.getElementType(); in verify()
91 VectorType aType = getLhsVectorType(); in verify() local
94 if (failed(verifyTileSize(*this, aType)) || in verify()
97 failed(verifyMultShape(*this, aType, bType, cType, 2))) in verify()
99 Type ta = aType.getElementType(); in verify()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/IR/ |
| H A D | NVGPUDialect.cpp | 123 Type aType = aVector.getElementType(); in verify() local
130 if (aType.isF64()) { in verify()
135 } else if (aType.isF32() || aType.isBF16() || aType.isF16() || in verify()
136 aType.isInteger(8) || aType.isInteger(4)) { in verify()
138 int operandBitwidth = aType.getIntOrFloatBitWidth(); in verify()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/AMX/Transforms/ |
| H A D | LegalizeForLLVMExport.cpp | 158 VectorType aType = op.getLhsVectorType(); in matchAndRewrite() local
163 getTileSizes(rewriter, *getTypeConverter(), aType, op.getLoc()); in matchAndRewrite()
180 VectorType aType = op.getLhsVectorType(); in matchAndRewrite() local
185 getTileSizes(rewriter, *getTypeConverter(), aType, op.getLoc()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/NVGPUToNVVM/ |
| H A D | NVGPUToNVVM.cpp | 266 auto aType = op.getMatrixA().getType().cast<VectorType>(); in matchAndRewrite() local
284 if (aType.getElementType().isInteger(8)) { in matchAndRewrite()
288 } else if (aType.getElementType().isInteger(4)) { in matchAndRewrite()
292 } else if (aType.getElementType().isF16()) { in matchAndRewrite()
295 } else if (aType.getElementType().isF64()) { in matchAndRewrite()
298 } else if (aType.getElementType().isF32()) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Utils/ |
| H A D | QuantUtils.cpp | 163 auto aType = a.getType().dyn_cast<ShapedType>(); in buildMatMulOpQuantizationAttr() local
166 if (!aType || !bType) in buildMatMulOpQuantizationAttr()
169 auto aQType = GET_UQTYPE(aType); in buildMatMulOpQuantizationAttr()
|
| /llvm-project-15.0.7/flang/include/flang/Evaluate/ |
| H A D | call.h | 99 if (const AssumedType * aType{std::get_if<AssumedType>(&u_)}) { in GetAssumedTypeDummy()
100 return &aType->symbol(); in GetAssumedTypeDummy()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/ |
| H A D | WmmaOpsToNvvm.cpp | 214 gpu::MMAMatrixType aType = in matchAndRewrite() local
216 ArrayRef<int64_t> aTypeShape = aType.getShape(); in matchAndRewrite()
224 NVVM::MMATypes sourceType = getElementType(aType); in matchAndRewrite()
|
| /llvm-project-15.0.7/flang/lib/Evaluate/ |
| H A D | intrinsics.cpp | 1664 if (std::optional<DynamicType> aType{sameArg->GetType()}) { in Match()
1665 if (result.categorySet.test(aType->category())) { in Match()
1666 resultType = *aType; in Match()
1668 resultType = DynamicType{*category, aType->kind()}; in Match()
|