-
Notifications
You must be signed in to change notification settings - Fork 3.4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[UnitTest][Metal] Parametrize allreduce GPU tests (#15749)
* [UnitTest][Metal] Parametrize allreduce GPU tests As a first step to addressing the Metal codegen errors that required the reversion in #15725, parametrizing the unit tests for `allreduce`. While these tests are parametrized with `@tvm.testing.parametrize_targets("cuda", "metal")`, the automatic `tvm.testing.requires_metal` marker inserted for the metal parametrization will cause them to be skipped if the metal runtime is unavailable, which includes the current CI. * Updated filename, device used when testing on metal
- Loading branch information
1 parent
67df20f
commit 64ab31e
Showing
1 changed file
with
57 additions
and
52 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters