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

..

external

examples of new GPT2 and JIT change (#2261)

2023-11-10 15:07:02 -05:00

extra

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

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

models

examples of new GPT2 and JIT change (#2261)

2023-11-10 15:07:02 -05:00

unit

external (#2191)

2023-10-31 13:57: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

helpers.py

perf: lazyop as dataclass (#1603)

2023-10-25 17:54:30 -04:00

test_allocators.py

For cuda get current free space from device, and retry alloc failures (#2197)

2023-11-09 15:53:50 -08:00

test_assign.py

move optimize_local_size (#2221)

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

test_conv_shapetracker.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

test_conv.py

fix wino mask upcast calculation (#2057)

2023-10-18 16:54:48 -07:00

test_custom_function.py

move all to compile api (#2203)

2023-11-01 23:01:32 -07:00

test_dtype.py

Internal cast 2 with more tests (#2257)

2023-11-10 10:42:39 -08:00

test_gc.py

fix up eye and fix gc test

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

test_jit.py

Fix mmaped in jit (#2225)

2023-11-06 14:54:21 -08:00

test_kernel_cache.py

move all to compile api (#2203)

2023-11-01 23:01:32 -07:00

test_lazybuffer.py

examples of new GPT2 and JIT change (#2261)

2023-11-10 15:07:02 -05:00

test_lazyop.py

Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)

2023-10-27 10:53:06 -10:00

test_linearizer_failures.py

one more fuzz linearizer failed example (#2260)

2023-11-10 09:17:37 -05:00

test_linearizer.py

examples of new GPT2 and JIT change (#2261)

2023-11-10 15:07:02 -05:00

test_net_speed.py

CI < 5 minutes (#1252)

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

test_nn.py

fix wino mask upcast calculation (#2057)

2023-10-18 16:54:48 -07:00

test_ops.py

less phi, proper phi (#2241)

2023-11-08 16:13:43 -08:00

test_optim.py

use class Foo: instead of class Foo(): (#1797)

2023-09-06 12:20:25 -07:00

test_randomness.py

Tensor.uniform set default to standard uniform (#2158)

2023-10-27 16:15:30 -04:00

test_sample.py

remove real_offset (#2234)

2023-11-07 17:30:53 -08:00

test_schedule.py

extract const if it's const (#2193)

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

test_search.py

simple runtime args (#2211)

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

test_specific_conv.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_speed_v_torch.py

move optimize_local_size (#2221)

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

test_symbolic_jit.py

examples of new GPT2 and JIT change (#2261)

2023-11-10 15:07:02 -05:00

test_symbolic_ops.py

Variable.bind newer (#2017)

2023-10-10 10:03:01 -07:00

test_symbolic_shapetracker.py

remove real_offset (#2234)

2023-11-07 17:30:53 -08:00

test_tensor.py

added from unaligned np test (#2134)

2023-10-23 11:38:57 -04:00

test_uops.py

simple runtime args (#2211)

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

test_webgpu.js

Enable Multi-Output Export (#2179)

2023-10-30 18:42:26 -07:00

test_winograd.py

print test durations and add speed (#2107)

2023-10-18 13:46:42 -07:00