Skip to content

Commit 10ec0d2

Browse files
authored
[MLIR] fix _f64ElementsAttr in ir.py (#91176)
1 parent 148b721 commit 10ec0d2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

mlir/python/mlir/ir.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ def _memref_type_attr(x, context):
274274
@register_attribute_builder("F64ElementsAttr")
275275
def _f64ElementsAttr(x, context):
276276
return DenseElementsAttr.get(
277-
np.array(x, dtype=np.int64),
277+
np.array(x, dtype=np.float64),
278278
type=F64Type.get(context=context),
279279
context=context,
280280
)

mlir/test/python/dialects/python_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ def attrBuilder():
167167
x_f32arr=[2.0, 3.0],
168168
x_f64=4.25, # CHECK-DAG: x_f64 = 4.250000e+00 : f64
169169
x_f64arr=[4.0, 8.0], # CHECK-DAG: x_f64arr = [4.000000e+00, 8.000000e+00]
170-
# CHECK-DAG: x_f64elems = dense<[3.952530e-323, 7.905050e-323]> : tensor<2xf64>
170+
# CHECK-DAG: x_f64elems = dense<[8.000000e+00, 1.600000e+01]> : tensor<2xf64>
171171
x_f64elems=[8.0, 16.0],
172172
# CHECK-DAG: x_flatsymrefarr = [@symbol1, @symbol2]
173173
x_flatsymrefarr=["symbol1", "symbol2"],

0 commit comments

Comments
 (0)