| .. |
|
external
|
ops_ext to replace cpu import (#3406)
|
2024-02-15 12:14:58 +01:00 |
|
imported
|
fix pad 0 size (#3277)
|
2024-01-30 08:58:10 -08:00 |
|
models
|
move graph.py and jit.py into features (#3376)
|
2024-02-12 17:34:34 +01:00 |
|
testextra
|
fix test extra issue (#3159)
|
2024-01-17 11:58:08 -08:00 |
|
unit
|
fix merging condition in merge_dims (#3363)
|
2024-02-12 11:50:26 +01:00 |
|
web
|
fast path for copy (#2548)
|
2023-12-01 11:34:47 -08: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
|
new lazy, benchmark (#2878)
|
2023-12-20 14:33:21 -08:00 |
|
test_assign.py
|
move dtypes to dtype.py (#2964)
|
2024-01-01 14:58:48 -08:00 |
|
test_conv.py
|
make WINO a context var, and LATEWINO in hlb_cifar (#3161)
|
2024-01-17 20:21:26 -05:00 |
|
test_conv_shapetracker.py
|
move create schedule and delete old API (#3377)
|
2024-02-12 18:10:45 +01:00 |
|
test_copy_speed.py
|
Disk shm refactor (#2912)
|
2023-12-22 09:23:37 -08:00 |
|
test_custom_function.py
|
move graph.py and jit.py into features (#3376)
|
2024-02-12 17:34:34 +01:00 |
|
test_device_speed.py
|
compiler support (#3260)
|
2024-01-26 23:36:40 -08:00 |
|
test_dtype.py
|
delete half_prekernel (#3388)
|
2024-02-14 15:40:48 +01:00 |
|
test_dtype_alu.py
|
move create schedule and delete old API (#3377)
|
2024-02-12 18:10:45 +01:00 |
|
test_fusion_op.py
|
move create schedule and delete old API (#3377)
|
2024-02-12 18:10:45 +01:00 |
|
test_gc.py
|
fix up eye and fix gc test
|
2023-02-27 06:53:18 -08:00 |
|
test_image_dtype.py
|
move dtypes to dtype.py (#2964)
|
2024-01-01 14:58:48 -08:00 |
|
test_jit.py
|
move graph.py and jit.py into features (#3376)
|
2024-02-12 17:34:34 +01:00 |
|
test_kernel_cache.py
|
move the compiler cache to be global (#2957)
|
2024-01-01 10:59:56 -08:00 |
|
test_lazybuffer.py
|
fix pad 0 size (#3277)
|
2024-01-30 08:58:10 -08:00 |
|
test_lazyop.py
|
move create schedule and delete old API (#3377)
|
2024-02-12 18:10:45 +01:00 |
|
test_linearizer.py
|
wmma: protect TC locals from modification and use only LOCAL (#3379)
|
2024-02-13 10:19:35 +01:00 |
|
test_linearizer_failures.py
|
wmma: protect TC locals from modification and use only LOCAL (#3379)
|
2024-02-13 10:19:35 +01:00 |
|
test_masked_st.py
|
multitensor start (#2676)
|
2023-12-07 17:07:05 -08:00 |
|
test_method_cache.py
|
add test that the compiler isn't used (#3025)
|
2024-01-05 17:24:01 -08:00 |
|
test_multitensor.py
|
move create schedule and delete old API (#3377)
|
2024-02-12 18:10:45 +01:00 |
|
test_net_speed.py
|
Remove pytest markers (#2831)
|
2023-12-18 18:53:28 -05:00 |
|
test_nn.py
|
make WINO a context var, and LATEWINO in hlb_cifar (#3161)
|
2024-01-17 20:21:26 -05:00 |
|
test_ops.py
|
cleanup atol in test_ops (#3368)
|
2024-02-10 19:44:44 -05:00 |
|
test_optim.py
|
remove realize from optimizer (#2880)
|
2023-12-20 16:42:41 -08:00 |
|
test_randomness.py
|
fix Tensor.randint ignoring kwargs (#3350)
|
2024-02-09 17:12:16 +01:00 |
|
test_sample.py
|
enable test_sample for all backend (#2593)
|
2023-12-03 17:20:27 -05:00 |
|
test_schedule.py
|
ops_ext to replace cpu import (#3406)
|
2024-02-15 12:14:58 +01:00 |
|
test_search.py
|
move create schedule and delete old API (#3377)
|
2024-02-12 18:10:45 +01:00 |
|
test_specific_conv.py
|
move dtypes to dtype.py (#2964)
|
2024-01-01 14:58:48 -08:00 |
|
test_speed_v_torch.py
|
move graph.py and jit.py into features (#3376)
|
2024-02-12 17:34:34 +01:00 |
|
test_symbolic_jit.py
|
move graph.py and jit.py into features (#3376)
|
2024-02-12 17:34:34 +01:00 |
|
test_symbolic_ops.py
|
fix gpt2 with empty prompt take 2 (#3102)
|
2024-01-12 14:46:36 -05:00 |
|
test_symbolic_shapetracker.py
|
unbind view or shapetracker also returns var_val (#3067)
|
2024-01-09 21:45:05 -05:00 |
|
test_tensor.py
|
Fix Tensor.mean to compute the mean correctly when 0-length axes are selected (#3318)
|
2024-02-05 01:40:37 -05:00 |
|
test_tensor_data.py
|
dtype fmt (#3132)
|
2024-01-15 11:31:54 -08:00 |
|
test_to_numpy.py
|
Apply ruff linting rules to tests (#2473)
|
2023-11-27 21:24:06 -08:00 |
|
test_uops.py
|
compiler support (#3260)
|
2024-01-26 23:36:40 -08:00 |
|
test_winograd.py
|
move create schedule and delete old API (#3377)
|
2024-02-12 18:10:45 +01:00 |
|
test_zero_copy.py
|
remove numpy from device (#3123)
|
2024-01-14 19:36:05 -08:00 |
91eb93f85a