Home
last modified time | relevance | path

Searched refs:elType (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DNvGpuSupport.cpp89 Type elType = type.vectorType.getElementType(); in inferTileWidthInBits() local
90 if (isAcc && elType.getIntOrFloatBitWidth() == 32) { in inferTileWidthInBits()
93 if (elType.getIntOrFloatBitWidth() == 64) { in inferTileWidthInBits()
104 Type elType = type.vectorType.getElementType(); in getMmaSyncRegisterType() local
105 if (elType.isF16()) { in getMmaSyncRegisterType()
113 if (elType.isF64()) { in getMmaSyncRegisterType()
122 if (elType.isInteger(8)) { in getMmaSyncRegisterType()
129 if (elType.isInteger(4)) { in getMmaSyncRegisterType()
136 if (elType.isInteger(32)) { in getMmaSyncRegisterType()
143 if (elType.isF32()) { in getMmaSyncRegisterType()
[all …]
H A DVectorToGPU.cpp436 Type elType = regInfo.registerLLVMType; in getMmaSyncVectorOperandType() local
437 if (auto vecType = elType.dyn_cast<VectorType>()) in getMmaSyncVectorOperandType()
438 elType = vecType.getElementType(); in getMmaSyncVectorOperandType()
439 return VectorType::get(shape, elType); in getMmaSyncVectorOperandType()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransforms.cpp2728 auto elType = x.getType().cast<VectorType>().getElementType(); in genOperator() local
2729 bool isInt = elType.isIntOrIndex(); in genOperator()
2811 auto elType = destType.getElementType(); in matchAndRewrite() local
2812 bool isInt = elType.isIntOrIndex(); in matchAndRewrite()
2816 VectorType resType = VectorType::get(destShape, elType); in matchAndRewrite()
2827 VectorType reductionType = VectorType::get(reductionShape, elType); in matchAndRewrite()