fix disktensor offset issue (#3532)

This commit is contained in:
George Hotz
2024-02-28 17:22:17 -08:00
committed by GitHub
parent 0b1fc5888a
commit 48918fa75a
2 changed files with 2 additions and 4 deletions

View File

@@ -205,7 +205,6 @@ class TestDiskTensor(unittest.TestCase):
helper_test_disk_tensor("dt5", [1,2,3,4,5], lambda x: x.reshape((1,5)))
helper_test_disk_tensor("dt6", [1,2,3,4], lambda x: x.reshape((2,2)))
@unittest.expectedFailure
def test_assign_to_different_dtype(self):
# NOTE: this is similar to Y_train in fetch_cifar
t = Tensor.empty(10, device=f'disk:{temp("dt7")}', dtype=dtypes.int64)