chenyu a753c8e071 examples of new GPT2 and JIT change (#2261) ...

* var_vals are global

* working with global ish

* better

* fix export model

* fix tests

* better kv cache

* does it run?

* use where for kvmask

* fix excessive var_vals

* fix import

* how does multigpu use this?

* llama kinda work

* faster and simpler

* cleanup

* fix conversation mode

* test cleanups

* fix one more test

* test cleanup

---------

Co-authored-by: George Hotz <geohot@gmail.com>

2023-11-10 15:07:02 -05: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

add world dataset (#2045)

2023-10-11 15:54:30 -07:00

disk

Fast DiskTensor to other Tensor (#916)

2023-06-03 12:25:41 -07:00

dist

feat: remove cache_id (#2236)

2023-11-08 08:09:21 -08:00

fastvits

use mad in opencl (#2198)

2023-11-01 10:40:08 -07:00

gemm

update metal matmul and matvec for compile api (#2238)

2023-11-08 08:08:35 -08:00

intel

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

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

optimization

merge kernel and optimizer (#2200)

2023-11-01 15:20:01 -07:00

ptx

ptx example

2023-05-26 19:28:51 -07:00

rocm

proper wmma (#2245)

2023-11-09 15:15:18 -08:00

triton

Disable ocelot cache + fix matvec in triton (#2010)

2023-10-17 10:33:32 -07: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

dump_cache.py

remove arm64, caching for cuda (#2201)

2023-11-01 18:44:00 -07:00

export_model.py

examples of new GPT2 and JIT change (#2261)

2023-11-10 15:07:02 -05: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

hip multigpu training (#1878)

2023-10-24 17:35:53 -04:00

introspection.py

move optimize_local_size (#2221)

2023-11-05 21:00:52 -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_ops.py

extract const if it's const (#2193)

2023-10-31 18:52:35 -07:00

onnx.py

Add ONNX to CI for other backends (#2069)

2023-10-17 09:33:54 -07:00

thneed.py

simple runtime args (#2211)

2023-11-03 12:31:29 -07:00

training.py

Replace (getenv("CI", "") != "") with helpers.CI (#2213)

2023-11-03 15:20:44 -07:00

utils.py

move optimize_local_size (#2221)

2023-11-05 21:00:52 -08:00