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.py	update pytest marks and CI test filters (#2587)	2023-12-03 15:20:44 -05:00
test_conv_shapetracker.py	fix some tests related to JitItem (#2279)	2023-11-11 23:00:35 -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.py	correct (dtype, device) in test_dtype.is_dtype_supported (#3007)	2024-01-04 00:25:37 -05:00
test_dtype_alu.py	move dtypes to dtype.py (#2964)	2024-01-01 14:58:48 -08: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.py	use float for acc dtype for half tensor sum	2024-01-04 13:12:22 -05:00
test_linearizer_failures.py	move dtypes to dtype.py (#2964)	2024-01-01 14:58:48 -08: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.py	move dtypes to dtype.py (#2964)	2024-01-01 14:58:48 -08:00
test_tensor_data.py	masked shapetrackers (#2657)	2023-12-06 11:22:26 -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