Home
last modified time | relevance | path

Searched full:scalar_fn (Results 1 – 9 of 9) sorted by relevance

/llvm-project/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Dscalar_expr.py52 return ScalarExpression(scalar_fn=self)
115 scalar_fn: Optional[ScalarFn] = None,
121 bool(scalar_fn) + bool(scalar_arg) + bool(scalar_const) + bool(scalar_index)
124 "One of 'scalar_fn', 'scalar_arg', 'scalar_const', or "
127 self.scalar_fn = scalar_fn
133 if self.scalar_fn:
134 scalar_fn_dict = dict(kind=self.scalar_fn.kind.name.lower())
135 if self.scalar_fn.fn_name:
136 scalar_fn_dict["fn_name"] = self.scalar_fn.fn_name
137 if self.scalar_fn.attr_name:
[all …]
H A Demitter.py371 elif expr.scalar_fn:
372 kind = expr.scalar_fn.kind.name.lower()
373 fn_name = expr.scalar_fn.fn_name
374 if expr.scalar_fn.attr_name:
375 fn_name, _ = self.fn_attr_mapping[expr.scalar_fn.attr_name]
378 self.expression(operand) for operand in expr.scalar_fn.operands
380 if expr.scalar_fn.kind == FunctionKind.TYPE:
381 operand_values = [expr.scalar_fn.type_var.name] + operand_values
/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml39 scalar_fn:
84 scalar_fn:
89 scalar_fn:
125 scalar_fn:
160 scalar_fn:
195 scalar_fn:
230 scalar_fn:
265 scalar_fn:
300 scalar_fn:
335 scalar_fn
[all...]
/llvm-project/mlir/test/python/dialects/linalg/opdsl/
H A Dassignments.py12 # CHECK: scalar_fn:
16 # CHECK: scalar_fn:
20 # CHECK: scalar_fn:
26 # CHECK: scalar_fn:
48 # CHECK: scalar_fn:
52 # CHECK: scalar_fn:
56 # CHECK: scalar_fn:
60 # CHECK: scalar_fn:
65 # CHECK: scalar_fn:
71 # CHECK: scalar_fn:
[all …]
/llvm-project/llvm/test/CodeGen/SPIRV/instructions/
H A Dscalar-bitwise-operations.ll13 ; CHECK-DAG: [[SCALAR_FN:%.+]] = OpTypeFunction [[SCALAR]] [[SCALAR]] [[SCALAR]]
24 ; CHECK: [[SCALAR_SHL]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
39 ; CHECK: [[SCALAR_LSHR]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
54 ; CHECK: [[SCALAR_ASHR]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
69 ; CHECK: [[SCALAR_AND]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
84 ; CHECK: [[SCALAR_OR]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
99 ; CHECK: [[SCALAR_XOR]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
H A Dscalar-integer-arithmetic.ll21 ; CHECK-DAG: [[SCALAR_FN:%.+]] = OpTypeFunction [[SCALAR]] [[SCALAR]] [[SCALAR]]
46 ; CHECK: [[SCALAR_ADD]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
74 ; CHECK: [[SCALAR_SUB]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
89 ; CHECK: [[SCALAR_MUL]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
104 ; CHECK: [[SCALAR_UDIV]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
119 ; CHECK: [[SCALAR_SDIV]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
H A Dscalar-floating-point-arithmetic.ll15 ; CHECK-DAG: [[SCALAR_FN:%.+]] = OpTypeFunction [[SCALAR]] [[SCALAR]] [[SCALAR]]
28 ; DISABLED-CHECK: [[SCALAR_FNEG]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
43 ; CHECK: [[SCALAR_FADD]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
58 ; CHECK: [[SCALAR_FSUB]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
73 ; CHECK: [[SCALAR_FMUL]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
88 ; CHECK: [[SCALAR_FDIV]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
103 ; CHECK: [[SCALAR_FREM]] = OpFunction [[SCALAR]] None [[SCALAR_FN]]
/llvm-project/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml42 scalar_fn:
47 scalar_fn:
55 scalar_fn:
231 scalar_fn:
290 scalar_fn:
295 scalar_fn:
368 scalar_fn:
/llvm-project/mlir/tools/mlir-linalg-ods-gen/
H A Dmlir-linalg-ods-yaml-gen.cpp269 /// - `scalar_fn`: A named function (see `ScalarFn`).
276 io.mapOptional("scalar_fn", info.scalarFn); in mapping()