proper wmma (#2245)

* proper wmma

* hip cast

* bugfixes

* bugfix

* that bug is fixed

---------

Co-authored-by: George Hotz <george@tinygrad.org>
This commit is contained in:
George Hotz
2023-11-09 15:15:18 -08:00
committed by GitHub
parent b7a31fb708
commit 80bf0b8586
6 changed files with 65 additions and 32 deletions

View File

@@ -27,13 +27,19 @@ repos:
pass_filenames: false
- id: tests
name: subset of (CPU) tests
entry: env CPU=1 pytest test/unit/ test/test_ops.py test/test_dtype.py test/test_schedule.py test/test_custom_function.py test/test_assign.py test/test_symbolic_shapetracker.py test/external/test_example.py
entry: env CPU=1 pytest test/unit/ test/test_ops.py test/test_dtype.py test/test_schedule.py test/test_custom_function.py test/test_assign.py test/test_symbolic_shapetracker.py
language: system
always_run: true
pass_filenames: false
- id: example
name: multi device tests
entry: python3 test/external/test_example.py
language: system
always_run: true
pass_filenames: false
- id: pylint
name: pylint
entry: python -m pylint tinygrad/
entry: python3 -m pylint tinygrad/
language: system
always_run: true
pass_filenames: false