github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix test_numpy_parity_and_backward_2d (#14755)

Browse Source 
test setup issue, test failed locally with `RUN_SLOW=1`
This commit is contained in:
chenyu
2026-02-14 17:59:00 -05:00
committed by GitHub
parent 043f5dbfa0
commit 902dc7c09c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/unit/test_indexing.py
View File

@@ -1000,7 +1000,7 @@ def assert_backward_eq(tensor: Tensor, indexer):
def get_set_tensor(indexed: Tensor, indexer):
set_size = indexed[indexer].shape
set_count = indexed[indexer].numel()
set_tensor = Tensor.randint(set_count, high=set_count).reshape(set_size) #.cast(dtypes.float64)
set_tensor = Tensor.randint(set_count, high=set_count).reshape(set_size).cast(indexed.dtype)
return set_tensor
@slow