chenyu ab7dfd637b use float for acc dtype for half tensor sum ...

we previously only upcast uint and int, and half was using half for acc.
change to acc in float for precision. but cast the result back to half to match torch/jax output dtype

2024-01-04 13:12:22 -05:00

..

external

removed redundant dtype hacks in onnx_ops (#2939)

2024-01-04 01:45:24 -05:00

extra

Remove pytest markers (#2831)

2023-12-18 18:53:28 -05:00

imported

new lazy, benchmark (#2878)

2023-12-20 14:33:21 -08:00

models

fix some long lines in tests (#3006)

2024-01-03 23:53:33 -05:00

unit

don't alloc for InterpretedASTRunner (#2999)

2024-01-03 17:05:53 -08: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_shapetracker.py

fix some tests related to JitItem (#2279)

2023-11-11 23:00:35 -05:00

test_conv.py

update pytest marks and CI test filters (#2587)

2023-12-03 15:20:44 -05:00

test_copy_speed.py

Disk shm refactor (#2912)

2023-12-22 09:23:37 -08:00

test_custom_function.py

search refactor (#2969)

2024-01-01 17:39:26 -08:00

test_dtype_alu.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

test_dtype.py

correct (dtype, device) in test_dtype.is_dtype_supported (#3007)

2024-01-04 00:25:37 -05:00

test_fusion_op.py

hotfix: compare on ids

2023-12-20 23:47:50 -08:00

test_gc.py

fix up eye and fix gc test

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

test_hip_rdna3.py

HIP cleanups (#2843)

2023-12-18 21:09:32 -08:00

test_image_dtype.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

test_jit.py

lazy tests (#2796)

2023-12-16 08:24:21 -08:00

test_kernel_cache.py

move the compiler cache to be global (#2957)

2024-01-01 10:59:56 -08:00

test_lazybuffer.py

canonicalize device in LazyBuffer constructor (#2991)

2024-01-03 12:55:25 -05:00

test_lazyop.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

test_linearizer_failures.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

test_linearizer.py

use float for acc dtype for half tensor sum

2024-01-04 13:12:22 -05:00

test_masked_st.py

multitensor start (#2676)

2023-12-07 17:07:05 -08:00

test_multitensor.py

canonicalize device in LazyBuffer constructor (#2991)

2024-01-03 12:55:25 -05:00

test_net_speed.py

Remove pytest markers (#2831)

2023-12-18 18:53:28 -05:00

test_nn.py

new lazy, benchmark (#2878)

2023-12-20 14:33:21 -08:00

test_ops.py

fix some long lines in tests (#3006)

2024-01-03 23:53:33 -05:00

test_optim.py

remove realize from optimizer (#2880)

2023-12-20 16:42:41 -08:00

test_randomness.py

fix some long lines in tests (#3006)

2024-01-03 23:53:33 -05:00

test_sample.py

enable test_sample for all backend (#2593)

2023-12-03 17:20:27 -05:00

test_schedule.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

test_search.py

new style device (#2530)

2023-11-30 17:07:16 -08: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 globalcounters to ops (#2960)

2024-01-01 14:21:02 -08:00

test_symbolic_jit.py

new lazy, benchmark (#2878)

2023-12-20 14:33:21 -08:00

test_symbolic_ops.py

new lazy, benchmark (#2878)

2023-12-20 14:33:21 -08:00

test_symbolic_shapetracker.py

tests from new lazy branch (#2774)

2023-12-14 23:06:39 -08:00

test_tensor_data.py

masked shapetrackers (#2657)

2023-12-06 11:22:26 -08:00

test_tensor.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

test_to_numpy.py

Apply ruff linting rules to tests (#2473)

2023-11-27 21:24:06 -08:00

test_uops.py

fix some long lines in tests (#3006)

2024-01-03 23:53:33 -05:00

test_winograd.py

get_lazyops() -> lazyops (#2884)

2023-12-20 18:04:49 -08:00

test_zero_copy.py

fix some long lines in tests (#3006)

2024-01-03 23:53:33 -05:00