executorch icon indicating copy to clipboard operation
executorch copied to clipboard

`bmm` and `baddbmm`

Open copyrightly opened this issue 9 months ago • 1 comments

Summary: We implement bmm. Thereafter baddbmm is also supported since it is decomposed as

aten.add.Tensor
aten.mul.Tensor
aten.mul.Tensor
aten.bmm.default

Differential Revision: D56679087

copyrightly avatar Apr 29 '24 18:04 copyrightly

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/3403

Note: Links to docs will display an error until the docs builds have been completed.

:white_check_mark: No Failures

As of commit 8263153c3c64c604a3b00ba69e2fe7689be2b3c1 with merge base 7fa91fdd6e37cffc73d7a5d76874d2bda2d9892f (image): :green_heart: Looks good so far! There are no failures yet. :green_heart:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

pytorch-bot[bot] avatar Apr 29 '24 18:04 pytorch-bot[bot]