Skip to content

Commit

Permalink
debugging mpi_type_and_elements_of w/ split 0
Browse files Browse the repository at this point in the history
  • Loading branch information
ClaudiaComito committed Dec 14, 2023
1 parent ba3fdbd commit 57621cb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions heat/core/tests/test_random.py
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,7 @@ def test_rand(self):
self.assertTrue((counts == 1).all())

# Two large arrays that were created after each other don't share any values
b = ht.random.rand(14, 7, 3, 12, 18, 42, split=4, dtype=ht.float64)
b = ht.random.rand(14, 7, 3, 12, 18, 42, split=0, dtype=ht.float64)
_, t_counts = torch.unique(b.larray, return_counts=True)
# Assert that no value appears more than once locally
print("DEBUGGING: t_counts>1: ", t_counts[t_counts > 1])
Expand All @@ -172,7 +172,7 @@ def test_rand(self):
print("DEBUGGING: np_b.shape: ", np_b.shape)
_, counts = np.unique(np_b, return_counts=True)
print("DEBUGGING: counts>1: ", counts > 1)
print("DEBUGGING: counts[counts > 1]: ", counts[counts > 1])
# print("DEBUGGING: counts[counts > 1]: ", counts[counts > 1])
print("DEBUGGING: b when counts > 1: ", _[counts > 1])
# Assert that no value appears more than once
self.assertTrue((counts == 1).all())
Expand Down

0 comments on commit 57621cb

Please sign in to comment.