Jeff Moe
|
661dcc5ed0
|
Reformat, uh, everything, with black
|
2023-12-04 22:01:04 -07:00 |
|
chenyu
|
28a67106ca
|
enable symbolic ops tests for hip (#2485)
|
2023-11-27 22:33:41 -08:00 |
|
chenyu
|
9eeba968cd
|
fix the variable arg order (#2382)
|
2023-11-21 12:02:31 -05:00 |
|
George Hotz
|
70a65c201e
|
JIT support in Interpreted (#2314)
* factor that out
* jit is supported everywhere
* fix some tests
* there's no jit supported device, the jit is everywhere
* fix test uops
|
2023-11-15 11:13:38 -08:00 |
|
chenyu
|
d86ea188dd
|
support symbolic shape in Interpreted (#2289)
* support symbolic shape in Interpreted
* simpler
* no InterpretedFlopCounter
* tragic NumNode
* regex is hard
|
2023-11-13 20:13:18 -05:00 |
|
chenyu
|
e2b83f1b42
|
Variable.bind newer (#2017)
* Variable.bind attempt 2
* ShapeTracker.unbind
* fix llama
* fix types
* test case
* View.vars cleanup
* include mask in symbolic source
* mask can be sint
* st.unbind in bufferops
* assert ast contain free Variable only
* cleanup
* conservative unbinding reduce op arg
* move reduceop unbind
* fix llama JIT arg behavior
|
2023-10-10 10:03:01 -07:00 |
|
Yixiang Gao
|
094d3d71be
|
with Tensor.train() (#1935)
* add with.train
* remove the rest TODOs
* fix pyflake
* fix pyflake error
* fix mypy
|
2023-09-28 18:02:31 -07:00 |
|
Gijs Koning
|
767bb35903
|
Enable symbolic ops tests for LLVM (#1898)
* Enable symbolic tests for HIP and LLVM
* Only llvm
|
2023-09-23 07:30:26 +08:00 |
|
chenyu
|
e67306ba04
|
symbolic shape type with TypeGuard (#1852)
|
2023-09-13 05:27:22 +08:00 |
|
chenyu
|
ebcda8a714
|
Move var_vals from ShapeTracker to LazyBuffer (#1819)
|
2023-09-08 09:25:10 -07:00 |
|
chenyu
|
66fbf4800b
|
fix symbolic_ops tests with Tensor.training=True (#1686)
|
2023-08-26 23:19:56 -04:00 |
|
chenyu
|
89e13f2f04
|
support symbols in shrink (#1611)
|
2023-08-22 09:08:21 -07:00 |
|
Yixiang Gao
|
8d6662a741
|
.cpu().numpy() -> .numpy() (#1594)
* .cpu().numpy() -> .numpy()
* restore ops_torch
* restore test_speed_v_torch
|
2023-08-21 09:53:29 -07:00 |
|
George Hotz
|
739f327d2d
|
Shorter (#1582)
* deleting lines
* remove insert dims
* if statement is never hit
* bug fixes
|
2023-08-20 08:12:16 -07:00 |
|
chenyu
|
dfec16cc83
|
Support arg int for CUDA kernel (#1565)
|
2023-08-18 09:19:40 -07:00 |
|
chenyu
|
11dd9b1741
|
symbolic codegen and exec (#1552)
* symbolic codegen and exec
* fix and add test
* no sketchy
* merge_dicts type
* dtypes._arg_int32
|
2023-08-16 14:43:41 -07:00 |