..
external
set metal fast math default to 0 (disabled) ( #3370 )
2024-02-14 11:42:33 +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
move create schedule and delete old API ( #3377 )
2024-02-12 18:10:45 +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