mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-07 21:13:41 +08:00
Daniel Koepping
12 lines
412 B
Python
12 lines
412 B
Python
import unittest
|
|
from tinygrad import Device, dtypes
|
|
from tinygrad.uop.ops import Ops
|
|
from test.runtime.test_uops import TestUOps
|
|
|
|
class TestFloatUOps(TestUOps):
|
|
@unittest.skipUnless(Device.DEFAULT == "PYTHON", "only python supports MULACC")
|
|
def test_mulacc(self):
|
|
self._test_top_fxn(Ops.MULACC, lambda a,b,c: a*b+c, (dtypes.float, dtypes.float, dtypes.float))
|
|
|
|
if __name__ == "__main__": unittest.main()
|