pytorch
c5b3727e - [JIT] OpInfo tests for nvfuser (#71299)

Commit
2 years ago
[JIT] OpInfo tests for nvfuser (#71299) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71299 These tests verify that for the same inputs, the eager version of an op and a traced, fused version of the op return the same output. Currently the tests don't check whether or not fusion actually occurred. Test Plan: Imported from OSS Reviewed By: eellison Differential Revision: D33595299 Pulled By: davidberard98 fbshipit-source-id: 26fdacf44941808c134953e7a883a02d13a43f19 (cherry picked from commit 8cd084e2e3130fcd5f9c99302d6d9bf4e21c25bb)
Author
Committer
Parents
Loading