..
external
codellama ( #1702 )
2023-09-02 08:45:12 -07:00
extra
[ready] Replacing os with pathlib ( #1708 )
2023-08-30 10:41:08 -07:00
models
faster gpt2 jit path and gpt2 in test_real_world ( #1738 )
2023-09-02 08:39:12 -07:00
unit
visitor pattern for sym_infer and unit tests ( #1733 )
2023-09-01 09:47:45 -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
allow zerosized tensors ( #1659 )
2023-08-30 10:39:24 -07:00
test_assign.py
Limit dims based on max size ( #1390 )
2023-07-31 19:18:19 -07:00
test_cache_collector.py
add cache collector ( #1595 )
2023-08-28 19:59:55 -07:00
test_conv.py
3x3 winograd convs ( #1675 )
2023-09-03 07:29:43 -07:00
test_conv_shapetracker.py
add cache collector ( #1595 )
2023-08-28 19:59:55 -07:00
test_custom_function.py
Optional: Reduce line count and simplify the LazyBuffer interface ( #1642 )
2023-08-22 21:01:10 -07:00
test_dtype.py
move device to ops ( #1646 )
2023-08-23 08:30:17 -07:00
test_gc.py
fix up eye and fix gc test
2023-02-27 06:53:18 -08:00
test_helpers.py
add dtypes test ( #1682 )
2023-08-28 08:12:15 -07:00
test_jit.py
support symbols in shrink ( #1611 )
2023-08-22 09:08:21 -07:00
test_lazybuffer.py
add cache collector ( #1595 )
2023-08-28 19:59:55 -07:00
test_linearizer.py
zero fold ( #1748 )
2023-09-03 13:48:11 -07:00
test_net_speed.py
CI < 5 minutes ( #1252 )
2023-07-23 13:00:56 -07:00
test_nn.py
3x3 winograd convs ( #1675 )
2023-09-03 07:29:43 -07:00
test_ops.py
Make __getitem__ TINYer ( #1661 )
2023-09-02 23:01:01 -04:00
test_optim.py
Add WEBGPU tests to CI ( #1463 )
2023-08-06 10:32:01 -07:00
test_randomness.py
Tensor.uniform with dtype=int bug fix ( #1593 )
2023-08-26 01:59:53 -04:00
test_specific_conv.py
move device to ops ( #1646 )
2023-08-23 08:30:17 -07:00
test_speed_v_torch.py
move device to ops ( #1646 )
2023-08-23 08:30:17 -07:00
test_symbolic_jit.py
llama JIT python runtime speedup ( #1633 )
2023-08-30 07:51:05 -07:00
test_symbolic_ops.py
fix symbolic_ops tests with Tensor.training=True ( #1686 )
2023-08-26 23:19:56 -04:00
test_symbolic_shapetracker.py
support symbols in shrink ( #1611 )
2023-08-22 09:08:21 -07:00
test_tensor.py
allow zerosized tensors ( #1659 )
2023-08-30 10:39:24 -07:00
test_uops.py
UnaryOps.NEG ( #1749 )
2023-09-03 12:44:26 -07:00
test_webgpu.js
Revert "update editorconfig, enforce via CI ( #1343 )" ( #1380 )
2023-07-31 10:35:50 -07:00