Skip to content

Commit

Permalink
Merge branch 'develop' into mlir-input-fuse-all-applicable
Browse files Browse the repository at this point in the history
  • Loading branch information
pfultz2 authored Nov 19, 2024
2 parents a12f765 + ed3037e commit f71650a
Show file tree
Hide file tree
Showing 4 changed files with 5 additions and 9 deletions.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -28,4 +28,4 @@ pybind/pybind11@3e9dfa2866941655c56877882565e7577de6fc7b --build
msgpack/[email protected] -DMSGPACK_BUILD_TESTS=Off
sqlite3@3.43.2 -DCMAKE_POSITION_INDEPENDENT_CODE=On
ROCm/composable_kernel@57cdd70b7cb14e5e3b60cd9a5f96ba8dc343763e -DCK_BUILD_JIT_LIB=On -DCMAKE_POSITION_INDEPENDENT_CODE=On
ROCm/rocMLIR@99fc9d24714ee7eae75ef8e414df4f2dacd3af16 -DBUILD_FAT_LIBROCKCOMPILER=On
ROCm/rocMLIR@45d830dbbc15fe84c41d95585f526a50719020eb -DBUILD_FAT_LIBROCKCOMPILER=On
2 changes: 1 addition & 1 deletion src/include/migraphx/op/dequantizelinear.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ struct dequantizelinear
{
MIGRAPHX_THROW("DEQUANTIZELINEAR: Zero point and input should be the same type.");
}
return {inputs[1].type(), inputs[0].lens(), inputs[0].strides()};
return inputs[0].with_lens(inputs[1].type(), inputs[0].lens());
}

argument compute(const shape& output_shape, std::vector<argument> args) const
Expand Down
8 changes: 2 additions & 6 deletions src/include/migraphx/op/quantizelinear.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -63,13 +63,9 @@ struct quantizelinear
}
if(inputs.size() == 3)
{
return {inputs[2].type(), inputs[0].lens(), inputs[0].strides()};
return inputs[0].with_lens(inputs[2].type(), inputs[0].lens());
}
if(out_type.has_value())
{
return {out_type.value(), inputs[0].lens(), inputs[0].strides()};
}
return {shape::uint8_type, inputs[0].lens(), inputs[0].strides()};
return inputs[0].with_lens(out_type.value_or(shape::uint8_type), inputs[0].lens());
}

argument compute(const shape& output_shape, std::vector<argument> args) const
Expand Down
2 changes: 1 addition & 1 deletion test/onnx/.onnxrt-commit
Original file line number Diff line number Diff line change
@@ -1 +1 @@
4d614e15bd9e6949bc3066754791da403e00d66c
ac9c135b9543ad0374fe335bc3dc5feb0f24f010

0 comments on commit f71650a

Please sign in to comment.