Skip to content

Fuse MatMulInteger and scale followed#4350

Merged
yufenglee merged 9 commits intomasterfrom
roli/matmul-integer
Jul 2, 2020
Merged

Fuse MatMulInteger and scale followed#4350
yufenglee merged 9 commits intomasterfrom
roli/matmul-integer

Conversation

@yufenglee
Copy link
Copy Markdown
Member

@yufenglee yufenglee commented Jun 26, 2020

Description: Describe your changes.
Fuse MatMulInteger + scale + bias followed
Add bias for DynamicQuantizeMatMul

@yufenglee yufenglee requested a review from a team as a code owner June 26, 2020 21:34
@yufenglee yufenglee requested a review from skottmckay June 29, 2020 22:15
.TypeConstraint(
"T1",
{"tensor(int8)", "tensor(uint8)"},
"Constrain input B data type to 8-bit integer tensor.")
Copy link
Copy Markdown
Contributor

@tracysh tracysh Jun 30, 2020

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is input A, right? Should the type constraints use the same "input 'A'" (single quoted) style as above? #Resolved

"of elements should be equal to the number of columns of input 'B'.",
"T2",
OpSchema::Optional)
.Input(6,
Copy link
Copy Markdown
Contributor

@tracysh tracysh Jun 30, 2020

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

run clang-format? the input seems to be indented oddly. should the 6 be on the next line to match the style from your other ops? #Resolved


const Tensor* a_scale_tensor = ctx->Input<Tensor>(2);
ORT_ENFORCE(IsScalarOr1ElementVector(a_scale_tensor),
"MatmulInteger : input A scale must be a scalar or 1D tensor of size 1. Per-Channel is not supported yet.");
Copy link
Copy Markdown
Contributor

@tracysh tracysh Jun 30, 2020

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

fixup operator name? #Resolved

@yufenglee yufenglee force-pushed the roli/matmul-integer branch from 092344a to 9961f58 Compare July 1, 2020 17:16
Copy link
Copy Markdown
Contributor

@skottmckay skottmckay left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

:shipit:

@yufenglee yufenglee merged commit 67a7d93 into master Jul 2, 2020
@yufenglee yufenglee deleted the roli/matmul-integer branch July 2, 2020 20:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants