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 |