Skip to content

Interchange Cast and Transpose operations to facilitate Transpose-MatMul fusion#6924

Merged
satyajandhyala merged 14 commits intothiagofc/ortmodule-apifrom
ortmodule-api/cast_tranpose_matmul_fusion
Mar 9, 2021
Merged

Interchange Cast and Transpose operations to facilitate Transpose-MatMul fusion#6924
satyajandhyala merged 14 commits intothiagofc/ortmodule-apifrom
ortmodule-api/cast_tranpose_matmul_fusion

Conversation

@satyajandhyala
Copy link
Contributor

Description: Modify Transpose-MatMul fusion code to interchange a Cast, between Transpose and MatMul preventing fusion, with the Transpose

Motivation and Context

@satyajandhyala satyajandhyala changed the base branch from master to thiagofc/ortmodule-api March 7, 2021 01:14
@satyajandhyala satyajandhyala changed the title [WIP] Interchange Cast and Transpose operations to facilitate Transpose-MatMul fusion Interchange Cast and Transpose operations to facilitate Transpose-MatMul fusion Mar 8, 2021
@satyajandhyala satyajandhyala marked this pull request as ready for review March 8, 2021 18:47
@satyajandhyala satyajandhyala requested a review from a team as a code owner March 8, 2021 18:47
@ashbhandare
Copy link
Contributor

could you also check in the python file that generated the onnx files for tests?

@satyajandhyala
Copy link
Contributor Author

could you also check in the python file that generated the onnx files for tests?

I actually created the ONNX file from the existing model

@ytaous
Copy link
Contributor

ytaous commented Mar 8, 2021

+1


In reply to: 793008584 [](ancestors = 793008584)

@ytaous
Copy link
Contributor

ytaous commented Mar 8, 2021

you should be able to take a similar .py code to generate the onnx model, it's fun to play with :-)


In reply to: 793034066 [](ancestors = 793034066)

@SherlockNoMad
Copy link
Contributor

Please add label to future PRs.
We are using the label to filter the PRs we should review.

…anspose+Cast+Matmul fusion to Cast+FusedMatMul.
@satyajandhyala satyajandhyala merged commit 48eebed into thiagofc/ortmodule-api Mar 9, 2021
@satyajandhyala satyajandhyala deleted the ortmodule-api/cast_tranpose_matmul_fusion branch March 9, 2021 16:54
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.

4 participants