add test_onnx_ops.py (#8569)

* boom

* fix webgpu

* use exact variable names in test so that AI can read easier

* add tag for specific test name like test a specific dtype

* fix ruff

* astype everything

* dtype in array creation

* just arange

* is 67% considered fixed?

* move test up

* small cleanups

* share function

* add qgemm as well

* add qgemm too

* make sure qgemm comes out as int

* take out qgemm for now

* fixed test

* add correct qgemm

* addressing feedback here too, early naive fix for now

* simplify bias and c to be minimalistic enough to test correctness

* refactored qlinearops

* maybe these asserts aren't the best..

* fix test

* updated tests to cover new ops

* try to add to CI

* move test_onnx_ops into testextra/

* more attention tests

* qlinear_add atol=1

* attention still not fullllllly correct

* it is what it is

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
This commit is contained in:
geohotstan
2025-02-25 05:15:22 +08:00
committed by GitHub
parent 56288243e6
commit f0b24d230c
5 changed files with 230 additions and 38 deletions

View File

@@ -54,6 +54,7 @@ setup(name='tinygrad',
"pillow",
"onnx==1.16.0",
"onnx2torch",
"onnxruntime",
"opencv-python",
"tabulate",
"tqdm",