Home
last modified time | relevance | path

Searched full:scalar_index (Results 1 – 10 of 10) sorted by relevance

/llvm-project/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Dscalar_expr.py95 return ScalarExpression(scalar_index=self)
118 scalar_index: Optional[ScalarIndex] = None,
121 bool(scalar_fn) + bool(scalar_arg) + bool(scalar_const) + bool(scalar_index)
125 "'scalar_index' must be specified"
130 self.scalar_index = scalar_index
147 elif self.scalar_index:
148 return dict(scalar_index=self.scalar_index.dim)
H A Demitter.py366 elif expr.scalar_index:
368 IntegerType.get_signless(64), expr.scalar_index.dim
/llvm-project/mlir/test/python/dialects/linalg/opdsl/
H A Dassignments.py100 # CHECK: scalar_index: 1
101 # CHECK: scalar_index: 0
/llvm-project/llvm/test/Transforms/CodeGenPrepare/X86/
H A Dgather-scatter-opt.ll37 define <4 x i32> @scalar_index(ptr %base, i64 %index) {
38 ; CHECK-LABEL: @scalar_index(
H A Dgather-scatter-opt-inseltpoison.ll38 define <4 x i32> @scalar_index(ptr %base, i64 %index) {
39 ; CHECK-LABEL: @scalar_index(
/llvm-project/llvm/test/Transforms/CodeGenPrepare/AArch64/
H A Dgather-scatter-opt.ll35 define <vscale x 4 x i32> @scalar_index(ptr %base, i64 %index, <vscale x 4 x i1> %mask) #0 {
36 ; CHECK-LABEL: @scalar_index(
H A Dgather-scatter-opt-inseltpoison.ll35 define <vscale x 4 x i32> @scalar_index(ptr %base, i64 %index, <vscale x 4 x i1> %mask) #0 {
36 ; CHECK-LABEL: @scalar_index(
/llvm-project/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml61 scalar_index: 1
/llvm-project/mlir/tools/mlir-linalg-ods-gen/
H A Dmlir-linalg-ods-yaml-gen.cpp268 /// - `scalar_index`: An iteration index.
275 io.mapOptional("scalar_index", info.index); in mapping()
/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml6732 scalar_index: 1
6755 scalar_index: 0