tinygrad/extra
David Hou 8e9db88474
expand after expr_idxs in Linearizer.global_load (#1818)
* small changes

* expand in terms of substitute, directly expand g_idxs g_valid

* delete expand_ops

* don't compare using hash

* any instead of in

thanks gijskoning

Co-authored-by: Gijs Koning <gijs-koning@live.nl>

* support tc

* testing code

* no more create_rednode

* maxsize none in view/node

* oops

* undo

* typing

* oops

* oops

* lmao

* lmao

* add expand multi test

* Node.iter_idxs

* type

* type

* delete checks!

* clean up a little?

* expand_idx in symbolic

* un-golf

* play around with types >.>

* test_substitute and also remove an incorrect test?

* get rid of range

* Update symbolic.py

* split out view cache change

* split out flat components change

* reduce diff

* reduce diff

* add some float4 tests

* fix

---------

Co-authored-by: Gijs Koning <gijs-koning@live.nl>
2023-09-29 10:33:34 -07:00
..
accel Make Triton work again (#1547) 2023-09-23 14:17:12 +08:00
assembly disable assembly (#1755) 2023-09-04 09:41:20 -07:00
datasets fixes (#1893) 2023-09-22 07:20:27 +08:00
disk Fast DiskTensor to other Tensor (#916) 2023-06-03 12:25:41 -07:00
dist add cache collector (#1595) 2023-08-28 19:59:55 -07:00
gemm wmma: add test and tensor core shape (#1925) 2023-09-28 18:04:28 -07:00
intel Revert "update editorconfig, enforce via CI (#1343)" (#1380) 2023-07-31 10:35:50 -07:00
ptx ptx example 2023-05-26 19:28:51 -07:00
rocm fast HIP gemm -> 100 TFLOPS (#1476) 2023-08-09 06:54:15 -07:00
triton A Triton backend for tinygrad (#470) 2023-02-01 11:53:57 -08:00
archprobe.py no werror in archprobe 2023-05-03 19:34:17 +00:00
augment.py [ready] Replacing os with pathlib (#1708) 2023-08-30 10:41:08 -07:00
export_model.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
gradcheck.py Fix: Jacobian tests [WIP] (#1126) 2023-07-05 15:36:22 -07:00
helpers.py helpers.py: improved test coverage + exception handling (#1165) 2023-07-07 10:26:05 -07:00
hip_wrapper.py HipGraph support (#1880) 2023-09-24 20:14:36 +08:00
introspection.py Devicebufferless (#708) 2023-03-18 14:40:23 -07:00
kernel_search.py move to shapetracker.py 2023-03-11 07:50:07 -08:00
lib_test_ast.py fix ast test 2023-03-04 07:49:25 -08:00
lr_scheduler.py Add pylint trailing whitespace rule (#1314) 2023-07-21 13:37:55 -04:00
onnx.py onnx full passing (#1076) 2023-09-05 13:23:32 -07:00
onnx_ops.py onnx full passing (#1076) 2023-09-05 13:23:32 -07:00
thneed.py add cache collector (#1595) 2023-08-28 19:59:55 -07:00
training.py with Tensor.train() (#1935) 2023-09-28 18:02:31 -07:00
utils.py expand after expr_idxs in Linearizer.global_load (#1818) 2023-09-29 10:33:34 -07:00