Commit b5e49114 authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

fixup! Add test case to parse in empty constant int64 proto buffer

parent 497262b4
......@@ -630,7 +630,7 @@ def constant_scalar_test():
@onnx_test
def constant_scalar_test2():
x = np.array([]).astype(np.int64)
y = helper.make_tensor_value_info('0', TensorProto.INT64, [1])
y = helper.make_tensor_value_info('0', TensorProto.INT64, [0])
node = onnx.helper.make_node(
'Constant',
......
......@@ -640,7 +640,7 @@ TEST_CASE(constant_scalar_test2)
{
migraphx::program p;
auto* mm = p.get_main_module();
mm->add_literal(migraphx::literal{migraphx::shape{migraphx::shape::int64_type, {1}}, {0}});
mm->add_literal(migraphx::literal{migraphx::shape{migraphx::shape::int64_type, {}}, {0}});
auto prog = optimize_onnx("constant_scalar_test2.onnx");
EXPECT(p == prog);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment