George Hotz 643cbdfd50 make embedding and GPT-2 fast (#1631) ...

* make embedding fast

* jit more, variable shape support

* print mem bw

2023-08-22 15:14:38 -07:00

..

external

make embedding and GPT-2 fast (#1631)

2023-08-22 15:14:38 -07:00

extra

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00

models

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00

unit

move state to nn/state (#1619)

2023-08-22 07:36:24 -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_allocators.py

init allocator for compiled backends (#1467)

2023-08-17 10:33:32 -07:00

test_assign.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_conv_shapetracker.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_conv.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_custom_function.py

flake8 (#1323)

2023-07-24 11:19:58 -04:00

test_dtype.py

Shorter (#1582)

2023-08-20 08:12:16 -07:00

test_gc.py

fix up eye and fix gc test

2023-02-27 06:53:18 -08:00

test_helpers.py

symbolic codegen and exec (#1552)

2023-08-16 14:43:41 -07:00

test_jit.py

support symbols in shrink (#1611)

2023-08-22 09:08:21 -07:00

test_lazybuffer.py

ShapeTracker.var_vals (#1540)

2023-08-14 18:53:37 -07:00

test_linearizer.py

CSE at uop level (#1483)

2023-08-19 23:40:40 -07:00

test_net_speed.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_nn.py

sparse cat cross entropy (#1597)

2023-08-21 14:14:54 -07:00

test_ops.py

add noalias to llvm (#1622)

2023-08-22 09:26:01 -07:00

test_optim.py

Add WEBGPU tests to CI (#1463)

2023-08-06 10:32:01 -07:00

test_randomness.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

test_specific_conv.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_speed_v_torch.py

add cpu if torch tensor (#1609)

2023-08-21 16:57:59 -07:00

test_symbolic_jit.py

support symbols in shrink (#1611)

2023-08-22 09:08:21 -07:00

test_symbolic_ops.py

support symbols in shrink (#1611)

2023-08-22 09:08:21 -07:00

test_symbolic_shapetracker.py

support symbols in shrink (#1611)

2023-08-22 09:08:21 -07:00

test_tensor.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

test_uops.py

make TestNonFloatUOps.test_mul_bool pass on all platforms (#1557)

2023-08-16 11:34:09 -07:00

test_webgpu.js

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00