tinygrad/test
Eli Frigo 801564f31b
Remove POW llop and add SQRT llop (#1104)
* fixed division by zero for fast operations

* made et closer to 0

* replace POW llop with SQRT

* updated mlops to swap SQRT and POW llops

* updated hlops to swap POW and SQRT

* added sqrt llop to cpu runtime

* added sqrt llop to cstyle codegen

* added POW llop to llvm ir codegen

* added SQRT llop to torch runtime

* moved pow from mlops to hlops

* found a better way to do reverse pow

* fixed indentation

* added SQRT llop to triton

* update docs to match new llops

* removed POW operator from assembly codegen

* added sqrt and rsqrt to pow hlop

* rewrote pow function in tensor.py

* Adjust tolerance

* Adjust for adamw

* Reduce for Adam too

* removed accidental leftover code

* removed all of accidental code

* added rsqrt test

* removed pow from mlops again

it was added back when resolving merge conflicts

---------

Co-authored-by: Jacky Lee <jla524@sfu.ca>
2023-07-05 18:07:58 -07:00
..
external Refactor nn.optim (#1091) 2023-07-02 15:07:30 -07:00
extra Make cross_process use cloudpickle (#1118) 2023-07-04 00:47:34 -07:00
models Refactor nn.optim (#1091) 2023-07-02 15:07:30 -07:00
unit from tensor cores + lb touchup (#1127) 2023-07-04 15:45:20 -07:00
__init__.py All devices are equal! (#196) 2020-12-15 23:44:08 -08:00
Dockerfile Docker fix (#1039) 2023-06-25 10:38:58 -07:00
test_assign.py Refactor LoadOps (#910) 2023-06-03 09:40:43 -07:00
test_conv.py Add conv binops_no_rerun test assertions (#665) 2023-03-10 19:09:48 -08:00
test_conv_shapetracker.py speed like you wouldn't believe (#626) 2023-03-02 07:49:19 -08:00
test_custom_function.py Optimizations in lazy.py (#987) 2023-06-26 13:55:42 -07:00
test_dtype.py Remove float64 (#1101) 2023-07-04 08:40:51 -07:00
test_gc.py fix up eye and fix gc test 2023-02-27 06:53:18 -08:00
test_jit.py assertions for jit 2023-05-05 21:56:32 -07:00
test_net_speed.py test speed llama (#1046) 2023-06-25 15:22:56 -07:00
test_nn.py Windows & Ubuntu CLANG CI support (#1011) 2023-06-19 09:33:24 -07:00
test_ops.py Remove POW llop and add SQRT llop (#1104) 2023-07-05 18:07:58 -07:00
test_optim.py Remove POW llop and add SQRT llop (#1104) 2023-07-05 18:07:58 -07:00
test_randomness.py Refactor LoadOps (#910) 2023-06-03 09:40:43 -07:00
test_specific_conv.py disable that test on LLVM. i have to stop pushing to master 2023-05-26 03:11:03 +00:00
test_speed_v_torch.py torch speed test, add add 2023-06-26 18:55:27 -07:00
test_tensor.py Fix: Jacobian tests [WIP] (#1126) 2023-07-05 15:36:22 -07:00