Skip to content

Commit

Permalink
Fix linters
Browse files Browse the repository at this point in the history
  • Loading branch information
alanwaketan committed Aug 23, 2023
1 parent 4ac20df commit 923de7e
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions test/spmd/test_xla_sharding_hlo.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@

import test_xla_sharding_base


class BasicShardingTest(test_xla_sharding_base.XlaShardingTest):

@classmethod
Expand All @@ -31,6 +32,7 @@ def test_xla_sharded_hlo_dump_post_optimizations(self):
if self.n_devices > 1:
self.assertIn('all-reduce', hlo)


if __name__ == '__main__':
test = unittest.main()
sys.exit(0 if test.result.wasSuccessful() else 1)

0 comments on commit 923de7e

Please sign in to comment.