mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
15 lines
642 B
Python
15 lines
642 B
Python
import ctypes, platform, unittest
|
|
from tinygrad import Device
|
|
from tinygrad.renderer.cstyle import ClangRenderer
|
|
|
|
class TestQCOM(unittest.TestCase):
|
|
# although part of the QCOM runtime, this tests flushing the CPU's dcache
|
|
@unittest.skipUnless(isinstance(Device["CPU"].renderer, ClangRenderer) and platform.machine().lower() in {"arm64", "aarch64"},
|
|
"dcache_flush's inline asm needs ClangRenderer, and runs on arm64")
|
|
def test_dcache_flush(self):
|
|
from tinygrad.runtime.ops_qcom import dcache_flush
|
|
buf = (ctypes.c_uint8 * 64)()
|
|
dcache_flush().fxn(buf, 0)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|