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

remove hip backend (#3783)

Browse Source 
* remove hip backend

* remove unused

* rhip

* more RHIP
This commit is contained in:
George Hotz
2024-03-17 10:12:16 -07:00
committed by GitHub
parent 2a14d1b5e0
commit 53adcb34f5
11 changed files with 33 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/unit/test_disk_tensor.py
View File

@@ -2,7 +2,7 @@ import pathlib, unittest
import numpy as np
from tinygrad import Tensor, Device, dtypes
from tinygrad.nn.state import safe_load, safe_save, get_state_dict, torch_load
from tinygrad.helpers import Timing, fetch, temp, getenv
from tinygrad.helpers import Timing, fetch, temp
from test.helpers import is_dtype_supported
def compare_weights_both(url):
@@ -214,7 +214,7 @@ class TestDiskTensor(unittest.TestCase):
np.testing.assert_array_equal(t.numpy(), np.array([3] * 10))
@unittest.skipIf(getenv("HIPCPU"), "no real HIP device exists in CI")
@unittest.skipIf(Device.DEFAULT == "RHIP", "no real HIP device exists in CI")
def test_bf16_disk_write_read(self):
t = Tensor([10000, -1, -1000, -10000, 20]).cast(dtypes.float32)
t.to(f"disk:{temp('f32')}").realize()