github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

amd mockgpu (#4535)

Browse Source 
* start mock amd gpu

* virt files

* cleaner

* init ci

* small fixes

* linter

* better?

* ugh

* linter

* fix

* diable some

* run shorter

* fixes

* add hcq test

* fix

* fix cmd revert
This commit is contained in:
nimlgen
2024-05-14 14:28:04 +03:00
committed by GitHub
parent 089eeec271
commit 2131556c2c
12 changed files with 741 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_randomness.py
View File

@@ -104,7 +104,7 @@ class TestRandomness(unittest.TestCase):
self.assertTrue(equal_distribution(Tensor.randn, torch.randn, lambda x: np.random.randn(*x)))
@given(strat.sampled_from([dtypes.float, dtypes.float16, dtypes.bfloat16]))
@unittest.skipIf(Device.DEFAULT in ["HSA", "RHIP"], "bfloat16 local buffer broken in HSA")
@unittest.skipIf(Device.DEFAULT in ["HSA", "RHIP", "AMD"], "bfloat16 local buffer broken in HSA")
def test_randn_finite(self, default_float):
if not is_dtype_supported(default_float): return
old_default_float = dtypes.default_float