| /llvm-project-15.0.7/mlir/lib/Dialect/LLVMIR/IR/ |
| H A D | FunctionCallUtils.cpp | 101 Type indexType) { in lookupOrCreateMallocFn() argument
103 moduleOp, kMalloc, indexType, in lookupOrCreateMallocFn()
108 Type indexType) { in lookupOrCreateAlignedAllocFn() argument
110 moduleOp, kAlignedAlloc, {indexType, indexType}, in lookupOrCreateAlignedAllocFn()
122 Type indexType) { in lookupOrCreateGenericAllocFn() argument
124 moduleOp, kGenericAlloc, indexType, in lookupOrCreateGenericAllocFn()
130 Type indexType) { in lookupOrCreateGenericAlignedAllocFn() argument
132 moduleOp, kGenericAlignedAlloc, {indexType, indexType}, in lookupOrCreateGenericAlignedAllocFn()
144 mlir::LLVM::lookupOrCreateMemRefCopyFn(ModuleOp moduleOp, Type indexType, in lookupOrCreateMemRefCopyFn() argument
148 ArrayRef<Type>{indexType, unrankedDescriptorType, unrankedDescriptorType}, in lookupOrCreateMemRefCopyFn()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/LLVMIR/ |
| H A D | FunctionCallUtils.h | 44 LLVM::LLVMFuncOp lookupOrCreateMallocFn(ModuleOp moduleOp, Type indexType);
46 Type indexType);
49 Type indexType);
51 Type indexType);
53 LLVM::LLVMFuncOp lookupOrCreateMemRefCopyFn(ModuleOp moduleOp, Type indexType,
|
| /llvm-project-15.0.7/mlir/lib/Conversion/LLVMCommon/ |
| H A D | MemRefBuilder.cpp | 26 indexType = value.getType() in MemRefDescriptor()
108 loc, indexType, value, in offset()
130 loc, indexType, value, in size()
136 auto indexPtrTy = LLVM::LLVMPointerType::get(indexType); in size()
137 auto arrayTy = LLVM::LLVMArrayType::get(indexType, rank); in size()
142 auto one = createIndexAttrConstant(builder, loc, indexType, 1); in size()
173 loc, indexType, value, in stride()
342 Type indexType = typeConverter.getIndexType(); in computeSizes() local
350 createIndexAttrConstant(builder, loc, indexType, in computeSizes()
371 loc, indexType, doubleRankIncremented, indexSize); in computeSizes()
[all …]
|
| H A D | Pattern.cpp | 251 Type indexType = getTypeConverter()->getIndexType(); in copyUnrankedDescriptors() local
257 mallocFunc = LLVM::lookupOrCreateMallocFn(module, indexType); in copyUnrankedDescriptors()
|
| H A D | TypeConverter.cpp | 479 auto indexType = IndexType::get(context); in promoteOneMemRefDescriptor() local
484 IntegerAttr::get(indexType, 1)); in promoteOneMemRefDescriptor()
|
| /llvm-project-15.0.7/mlir/test/python/dialects/ |
| H A D | tensor.py | 21 indexType = IndexType.get()
33 c0 = arith.ConstantOp(indexType, 0)
34 c1 = arith.ConstantOp(indexType, 1)
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/ |
| H A D | MemoryPromotion.cpp | 60 auto indexType = b.getIndexType(); in insertCopyLoops() local
63 threadIds.push_back(b.create<gpu::ThreadIdOp>(indexType, dim)); in insertCopyLoops()
64 blockDims.push_back(b.create<gpu::BlockDimOp>(indexType, dim)); in insertCopyLoops()
|
| H A D | AllReduceLowering.cpp | 35 indexType(IndexType::get(reduceOp.getContext())), in GpuAllReduceRewriter()
113 Value index = create<arith::IndexCastOp>(indexType, subgroupId); in rewrite()
130 Value index = create<arith::IndexCastOp>(indexType, invocationIdx); in rewrite()
154 Value dim = create<T>(indexType, dimension); in getDimOp()
384 Type indexType; member
|
| /llvm-project-15.0.7/flang/lib/Optimizer/Builder/ |
| H A D | DoLoopHelper.cpp | 43 auto indexType = builder.getIndexType(); in createLoop() local
44 auto zero = builder.createIntegerConstant(loc, indexType, 0); in createLoop()
|
| H A D | Character.cpp | 133 auto indexType = builder.getIndexType(); in toExtendedValue() local
138 builder.createIntegerConstant(loc, indexType, extent)); in toExtendedValue()
524 auto indexType = builder.getIndexType(); in createLenTrim() local
525 auto len = builder.createConvert(loc, indexType, str.getLen()); in createLenTrim()
526 auto one = builder.createIntegerConstant(loc, indexType, 1); in createLenTrim()
527 auto minusOne = builder.createIntegerConstant(loc, indexType, -1); in createLenTrim()
528 auto zero = builder.createIntegerConstant(loc, indexType, 0); in createLenTrim()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUToSPIRV/ |
| H A D | GPUToSPIRV.cpp | 134 auto indexType = typeConverter->getIndexType(); in matchAndRewrite() local
138 spirv::getBuiltinVariableValue(op, builtin, indexType, rewriter); in matchAndRewrite()
140 op, indexType, spirvBuiltin, in matchAndRewrite()
151 auto indexType = typeConverter->getIndexType(); in matchAndRewrite() local
154 spirv::getBuiltinVariableValue(op, builtin, indexType, rewriter); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/LinalgToSPIRV/ |
| H A D | LinalgToSPIRV.cpp | 140 auto indexType = typeConverter->getIndexType(); in matchAndRewrite() local
143 Value x = getLocalInvocationDimSize(genericOp, /*dim=*/0, indexType, loc, in matchAndRewrite()
168 Value zero = spirv::ConstantOp::getZero(indexType, loc, rewriter); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TensorToSPIRV/ |
| H A D | TensorToSPIRV.cpp | 81 auto indexType = typeConverter.getIndexType(); in matchAndRewrite() local
84 /*offset=*/0, indexType, loc, rewriter); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/Transforms/ |
| H A D | SPIRVConversion.cpp | 686 Type indexType) { in getPushConstantStorageType() argument
687 auto arrayType = spirv::ArrayType::get(indexType, elementCount, in getPushConstantStorageType()
723 Type indexType) { in getOrInsertPushConstantVariable() argument
728 auto type = getPushConstantStorageType(elementCount, builder, indexType); in getOrInsertPushConstantVariable()
797 auto indexType = typeConverter.getIndexType(); in getElementPtr() local
800 auto zero = spirv::ConstantOp::getZero(indexType, loc, builder); in getElementPtr()
809 linearizeIndex(indices, strides, offset, indexType, loc, builder)); in getElementPtr()
|
| H A D | LowerABIAttributesPass.cpp | 185 auto indexType = typeConverter.getIndexType(); in matchAndRewrite() local
217 spirv::ConstantOp::getZero(indexType, funcOp.getLoc(), rewriter); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/test/CAPI/ |
| H A D | ir.c | 118 MlirType indexType = in makeAndDumpAdd() local
127 mlirOperationStateAddResults(&constZeroState, 1, &indexType); in makeAndDumpAdd()
138 mlirOperationStateAddResults(&dimState, 1, &indexType); in makeAndDumpAdd()
144 mlirBlockAddArgument(loopBody, indexType, location); in makeAndDumpAdd()
154 mlirOperationStateAddResults(&constOneState, 1, &indexType); in makeAndDumpAdd()
1673 MlirType indexType = mlirIndexTypeGet(ctx); in testOperands() local
1683 mlirOperationStateAddResults(&constZeroState, 1, &indexType); in testOperands()
1695 mlirOperationStateAddResults(&constOneState, 1, &indexType); in testOperands()
1742 MlirType indexType = mlirIndexTypeGet(ctx); in testClone() local
1751 mlirOperationStateAddResults(&constZeroState, 1, &indexType); in testClone()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Conversion/LLVMCommon/ |
| H A D | MemRefBuilder.h | 83 Type getIndexType() { return indexType; }; in getIndexType()
111 Type indexType;
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/ |
| H A D | TestLinalgTransforms.cpp | 255 Type indexType = b.getIndexType(); in getGpuProcIds() local
258 procInfo[count - 1 - i] = {b.create<IdOp>(loc, indexType, dim), in getGpuProcIds()
259 b.create<NProcsOp>(loc, indexType, dim)}; in getGpuProcIds()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 1241 auto indexType = parser.getBuilder().getIndexType(); in parse() local
1289 if (parser.resolveOperands(strideInfo, indexType, result.operands)) in parse()
1380 auto indexType = parser.getBuilder().getIndexType(); in parse() local
2805 auto indexType = builder.getIndexType(); in parseAffineMinMaxOp() local
2816 parser.resolveOperands(dimInfos, indexType, result.operands) || in parseAffineMinMaxOp()
2817 parser.resolveOperands(symInfos, indexType, result.operands) || in parseAffineMinMaxOp()
2818 parser.addTypeToList(indexType, result.types)); in parseAffineMinMaxOp()
3597 Type indexType = parser.getBuilder().getIndexType(); in deduplicateAndResolveOperands() local
3600 if (parser.resolveOperands(list, indexType, valueOperands)) in deduplicateAndResolveOperands()
3740 auto indexType = builder.getIndexType(); in parse() local
[all …]
|
| /llvm-project-15.0.7/flang/lib/Optimizer/CodeGen/ |
| H A D | CodeGen.cpp | 185 auto idxTy = lowerTy().indexType(); in loadStrideFromBox()
362 mlir::Type ity = lowerTy().indexType(); in matchAndRewrite()
1003 indexType, in getMalloc()
1031 auto ity = lowerTy().indexType(); in matchAndRewrite()
1257 loc, this->lowerTy().indexType(), gep);
1731 mlir::Type idxTy = lowerTy().indexType(); in matchAndRewrite()
1818 mlir::Type idxTy = lowerTy().indexType(); in sliceBox()
1919 mlir::Type idxTy = lowerTy().indexType(); in reshapeBox()
2146 mlir::Type idxTy = lowerTy().indexType(); in doRewrite()
2443 auto idxTy = lowerTy().indexType(); in doRewriteBox()
[all …]
|
| H A D | TypeConverter.h | 161 mlir::Type indexType() { return mlir::IntegerType::get(&getContext(), 64); } in indexType() function
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 1024 auto indexType = parser.getBuilder().getIndexType(); in parse() local
1055 parser.resolveOperands(srcIndexInfos, indexType, result.operands) || in parse()
1057 parser.resolveOperands(dstIndexInfos, indexType, result.operands) || in parse()
1059 parser.resolveOperand(numElementsInfo, indexType, result.operands) || in parse()
1062 parser.resolveOperands(tagIndexInfos, indexType, result.operands)) in parse()
1066 if (parser.resolveOperands(strideInfo, indexType, result.operands)) in parse()
1212 Type indexType = parser.getBuilder().getIndexType(); in parse() local
1217 parser.resolveOperands(ivs, indexType, result.operands)) in parse()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1188 Type indexType = getIndexType(); in convertSourceMemRefToDescriptor() local
1189 if (dimSize.getType() != indexType) in convertSourceMemRefToDescriptor()
1191 rewriter, loc, indexType, dimSize); in convertSourceMemRefToDescriptor()
1261 Type indexType = getTypeConverter()->getIndexType(); in convertSourceMemRefToDescriptor() local
1265 {indexType, indexType}, {loc, loc}); in convertSourceMemRefToDescriptor()
1288 Type llvmIndexPtrType = LLVM::LLVMPointerType::get(indexType); in convertSourceMemRefToDescriptor()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizationOps.cpp | 384 Type indexType = parser.getBuilder().getIndexType(); in parse() local
385 if (parser.resolveOperands(dynamicSizesOperands, indexType, result.operands)) in parse()
|
| /llvm-project-15.0.7/flang/lib/Optimizer/Dialect/ |
| H A D | FIROps.cpp | 1670 auto indexType = builder.getIndexType(); in parse() local
1673 parser.resolveOperand(lb, indexType, result.operands) || in parse()
1675 parser.resolveOperand(ub, indexType, result.operands) || in parse()
1678 parser.resolveOperand(step, indexType, result.operands) || in parse()
1739 argTypes.push_back(indexType); in parse()
1979 auto indexType = builder.getIndexType(); in parse() local
1981 parser.resolveOperand(lb, indexType, result.operands) || in parse()
1983 parser.resolveOperand(ub, indexType, result.operands) || in parse()
1985 parser.resolveOperand(step, indexType, result.operands)) in parse()
2015 result.types.push_back(indexType); in parse()
[all …]
|