Home
last modified time | relevance | path

Searched refs:dims (Results 1 – 25 of 106) sorted by relevance

12345

/llvm-project-15.0.7/polly/utils/pyscop/
H A Dpyscop.py34 dims = ['d' + str(i) for i in range(numberDimensions)]
35 dimString = ",".join(dims)
37 changedDims = dims
38 first = dims[dimensionA]
39 second = dims[dimensionB]
59 dims = ['d' + str(i) for i in range(numberDimensions)]
60 dimString = ",".join(dims)
62 changedDims = dims
63 smd = dims[stripMineDim]
/llvm-project-15.0.7/mlir/lib/IR/
H A DTypeUtilities.cpp61 for (auto dims : llvm::zip(shape1, shape2)) { in verifyCompatibleShape() local
62 int64_t dim1 = std::get<0>(dims); in verifyCompatibleShape()
63 int64_t dim2 = std::get<1>(dims); in verifyCompatibleShape()
103 LogicalResult mlir::verifyCompatibleDims(ArrayRef<int64_t> dims) { in verifyCompatibleDims() argument
104 if (dims.empty()) in verifyCompatibleDims()
107 dims.begin(), dims.end(), dims.front(), [](auto fold, auto dim) { in verifyCompatibleDims()
110 return success(llvm::all_of(dims, [&](auto dim) { in verifyCompatibleDims()
156 auto dims = llvm::to_vector<8>(llvm::map_range( in verifyCompatibleShapes() local
160 if (verifyCompatibleDims(dims).failed()) in verifyCompatibleShapes()
/llvm-project-15.0.7/openmp/runtime/test/worksharing/for/
H A Dkmp_doacross_check.c27 struct dim dims; in main() local
30 dims.lo = 1; in main()
31 dims.up = N-1; in main()
32 dims.st = 1; in main()
38 __kmpc_doacross_init(NULL,gtid,1,&dims); // thread starts the loop in main()
/llvm-project-15.0.7/mlir/lib/Dialect/LLVMIR/IR/
H A DLLVMTypeSyntax.cpp257 SmallVector<int64_t, 2> dims; in parseVectorType() local
262 parser.parseDimensionList(dims, /*allowDynamic=*/true) || in parseVectorType()
271 if (dims.empty() || dims.size() > 2 || in parseVectorType()
272 ((dims.size() == 2) ^ (dims[0] == -1)) || in parseVectorType()
273 (dims.size() == 2 && dims[1] == -1)) { in parseVectorType()
279 bool isScalable = dims.size() == 2; in parseVectorType()
287 return parser.getChecked<LLVMFixedVectorType>(loc, elementType, dims[0]); in parseVectorType()
293 SmallVector<int64_t, 1> dims; in parseArrayType() local
298 parser.parseDimensionList(dims, /*allowDynamic=*/false) || in parseArrayType()
302 if (dims.size() != 1) { in parseArrayType()
[all …]
/llvm-project-15.0.7/flang/test/Fir/
H A Daffine-promotion.fir12 %dims = fir.shape %len : (index) -> !fir.shape<1>
17 %a1_idx = fir.array_coor %a1(%dims) %i
21 %a2_idx = fir.array_coor %a2(%dims) %i
26 %t1_idx = fir.array_coor %t1(%dims) %i
32 %t1_idx = fir.array_coor %t1(%dims) %i
36 %a2_idx = fir.array_coor %a2(%dims) %i
41 %a3_idx = fir.array_coor %a3(%dims) %i
86 %dims = fir.shape %len : (index) -> !fir.shape<1>
94 %a_idx = fir.array_coor %a(%dims) %i
98 %aj_idx = fir.array_coor %a(%dims) %j
[all …]
/llvm-project-15.0.7/flang/lib/Evaluate/
H A Dconstant.cpp142 const ConstantSubscripts &dims) const -> std::vector<Element> { in Reshape()
143 std::size_t n{TotalElementCount(dims)}; in Reshape()
178 auto Constant<T>::Reshape(ConstantSubscripts &&dims) const -> Constant { in Reshape()
179 return {Base::Reshape(dims), std::move(dims)}; in Reshape()
267 ConstantSubscripts &&dims) const -> Constant<Result> { in Reshape()
268 std::size_t n{TotalElementCount(dims)}; in Reshape()
280 return {length_, std::move(elements), std::move(dims)}; in Reshape()
349 auto Constant<SomeDerived>::Reshape(ConstantSubscripts &&dims) const in Reshape()
351 return {result().derivedTypeSpec(), Base::Reshape(dims), std::move(dims)}; in Reshape()
/llvm-project-15.0.7/mlir/test/Conversion/SCFToGPU/
H A Dno_blocks_no_threads.mlir1 …pt -pass-pipeline="func.func(convert-affine-for-to-gpu{gpu-block-dims=0 gpu-thread-dims=1})" %s | …
2 …pt -pass-pipeline="func.func(convert-affine-for-to-gpu{gpu-block-dims=1 gpu-thread-dims=0})" %s | …
H A Dstep_one.mlir1 …pt -pass-pipeline="func.func(convert-affine-for-to-gpu{gpu-block-dims=1 gpu-thread-dims=1})" %s | …
2 …pt -pass-pipeline="func.func(convert-affine-for-to-gpu{gpu-block-dims=2 gpu-thread-dims=2})" %s | …
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h114 dims = llvm::to_vector<4>(shapedType.getShape()); in ShapedTypeComponents()
126 : dims(std::forward<Arg>(arg)), elementType(elementType), attr(attr), in dims() function
130 : dims(vec.begin(), vec.end()), elementType(elementType), attr(attr),
137 return dims; in getDims()
152 ShapeStorageT dims; variable
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp93 SmallVector<unsigned> dims; in splitReduction() local
94 op.getReductionDims(dims); in splitReduction()
95 assert(dims.size() == 1); in splitReduction()
96 unsigned reductionDim = dims[0]; in splitReduction()
284 SmallVector<unsigned> dims; in splitReductionByScaling() local
285 op.getReductionDims(dims); in splitReductionByScaling()
286 if (dims.empty()) in splitReductionByScaling()
289 unsigned reductionDimPos = dims[0]; in splitReductionByScaling()
339 SmallVector<Value> dims = in splitReductionByScaling() local
344 b.create<bufferization::AllocTensorOp>(loc, newT, dims); in splitReductionByScaling()
[all …]
/llvm-project-15.0.7/mlir/unittests/Dialect/Affine/Analysis/
H A DAffineStructuresParserTest.cpp27 unsigned dims, unsigned syms, ArrayRef<SmallVector<int64_t, 4>> ineqs, in makeFACFromConstraints() argument
30 IntegerPolyhedron fac(ineqs.size(), eqs.size(), dims + syms + 1,
31 PresburgerSpace::getSetSpace(dims, syms, 0));
/llvm-project-15.0.7/flang/test/Lower/Intrinsics/
H A Dreshape.f9080 integer :: dims(4) = [2,2,2,2] local
81 call some_proc(reshape(tmp, dims(i:2)))
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPU.cpp160 SmallVector<Value, 6> dims; member
185 dims.reserve(numLoops); in collectBounds()
202 dims.push_back(range); in collectBounds()
229 Value gridSizeX = numBlockDims > 0 ? dims[0] : constOne; in createLaunch()
230 Value gridSizeY = numBlockDims > 1 ? dims[1] : constOne; in createLaunch()
231 Value gridSizeZ = numBlockDims > 2 ? dims[2] : constOne; in createLaunch()
232 Value blockSizeX = numThreadDims > 0 ? dims[numBlockDims] : constOne; in createLaunch()
233 Value blockSizeY = numThreadDims > 1 ? dims[numBlockDims + 1] : constOne; in createLaunch()
234 Value blockSizeZ = numThreadDims > 2 ? dims[numBlockDims + 2] : constOne; in createLaunch()
/llvm-project-15.0.7/mlir/examples/toy/Ch4/include/toy/
H A DAST.h80 std::vector<int64_t> dims; variable
84 std::vector<int64_t> dims) in LiteralExprAST() argument
86 dims(std::move(dims)) {} in LiteralExprAST()
89 llvm::ArrayRef<int64_t> getDims() { return dims; } in getDims()
/llvm-project-15.0.7/mlir/examples/toy/Ch2/include/toy/
H A DAST.h80 std::vector<int64_t> dims; variable
84 std::vector<int64_t> dims) in LiteralExprAST() argument
86 dims(std::move(dims)) {} in LiteralExprAST()
89 llvm::ArrayRef<int64_t> getDims() { return dims; } in getDims()
/llvm-project-15.0.7/mlir/examples/toy/Ch3/include/toy/
H A DAST.h80 std::vector<int64_t> dims; variable
84 std::vector<int64_t> dims) in LiteralExprAST() argument
86 dims(std::move(dims)) {} in LiteralExprAST()
89 llvm::ArrayRef<int64_t> getDims() { return dims; } in getDims()
/llvm-project-15.0.7/mlir/examples/toy/Ch1/include/toy/
H A DAST.h80 std::vector<int64_t> dims; variable
84 std::vector<int64_t> dims) in LiteralExprAST() argument
86 dims(std::move(dims)) {} in LiteralExprAST()
89 llvm::ArrayRef<int64_t> getDims() { return dims; } in getDims()
/llvm-project-15.0.7/mlir/examples/toy/Ch5/include/toy/
H A DAST.h80 std::vector<int64_t> dims; variable
84 std::vector<int64_t> dims) in LiteralExprAST() argument
86 dims(std::move(dims)) {} in LiteralExprAST()
89 llvm::ArrayRef<int64_t> getDims() { return dims; } in getDims()
/llvm-project-15.0.7/mlir/examples/toy/Ch6/include/toy/
H A DAST.h80 std::vector<int64_t> dims; variable
84 std::vector<int64_t> dims) in LiteralExprAST() argument
86 dims(std::move(dims)) {} in LiteralExprAST()
89 llvm::ArrayRef<int64_t> getDims() { return dims; } in getDims()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Utils/
H A DMerger.h157 hasSparseOut(false), dims(t + 1, std::vector<Dim>(l, Dim::kUndef)) {} in Merger()
237 return dims[t][i] == d; in isDim()
250 void setDim(unsigned t, unsigned i, Dim d) { dims[t][i] = d; } in setDim()
299 std::vector<std::vector<Dim>> dims; variable
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DInferTypeOpInterface.cpp65 getDims(res.dims); in getDims()
137 SmallVector<int64_t> dims; in dump() local
138 getDims(dims); in dump()
139 auto mapped = llvm::map_range(dims, [](int64_t dim) -> std::string { in dump()
/llvm-project-15.0.7/mlir/test/mlir-tblgen/
H A Dattrdefs.td61 ArrayRefParameter<"int", "Matrix dimensions">:$dims,
69 …SimpleTypeA exampleTdType, ::llvm::APFloat apFloat, ::llvm::ArrayRef<int> dims, ::mlir::Type inner…
70 …SimpleTypeA exampleTdType, ::llvm::APFloat apFloat, ::llvm::ArrayRef<int> dims, ::mlir::Type inner…
91 // DEF-SAME: (dims == std::get<3>(tblgenKey)) &&
96 // DEF-SAME: CompoundAAttrStorage(widthOfSomething, exampleTdType, apFloat, dims, inner);
/llvm-project-15.0.7/mlir/examples/toy/Ch7/include/toy/
H A DAST.h82 std::vector<int64_t> dims; variable
86 std::vector<int64_t> dims) in LiteralExprAST() argument
88 dims(std::move(dims)) {} in LiteralExprAST()
91 llvm::ArrayRef<int64_t> getDims() { return dims; } in getDims()
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Dcomprehension.py503 def __getitem__(self, dims: Sequence[AffineExprDef]) -> TensorUse:
508 if not isinstance(dims, tuple):
509 dims = (dims,) # Handle single subscript case.
511 if dims == (None,):
512 dims = ()
515 for expr_def in dims:
522 def __setitem__(self, dims: Sequence[AffineExprDef], value: TensorExpression):
531 use = self[dims]
/llvm-project-15.0.7/openmp/runtime/src/
H A Dkmp_csupport.cpp4102 range_length = dims[j].up - dims[j].lo + 1; in __kmpc_doacross_init()
4104 if (dims[j].st > 0) { in __kmpc_doacross_init()
4105 KMP_DEBUG_ASSERT(dims[j].up > dims[j].lo); in __kmpc_doacross_init()
4106 range_length = (kmp_uint64)(dims[j].up - dims[j].lo) / dims[j].st + 1; in __kmpc_doacross_init()
4108 KMP_DEBUG_ASSERT(dims[j].lo > dims[j].up); in __kmpc_doacross_init()
4110 (kmp_uint64)(dims[j].lo - dims[j].up) / (-dims[j].st) + 1; in __kmpc_doacross_init()
4122 trace_count = dims[0].up - dims[0].lo + 1; in __kmpc_doacross_init()
4124 KMP_DEBUG_ASSERT(dims[0].up > dims[0].lo); in __kmpc_doacross_init()
4125 trace_count = (kmp_uint64)(dims[0].up - dims[0].lo) / dims[0].st + 1; in __kmpc_doacross_init()
4127 KMP_DEBUG_ASSERT(dims[0].lo > dims[0].up); in __kmpc_doacross_init()
[all …]

12345