chenyu 26200574dc load_state_dict test cases when model and data shard differently (#7774) ... current behavior is weird... when model is sharded and state_dict is not, load shards the state_dict and model shard axis does not change. but if model and state_dict are sharded differently, model shard axis becomes the state_dict axis after load. it should either always use model shard axis or always use state_dict shard		2024-11-18 16:08:24 -05:00
..
external	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
imported	node cleanup + local metal test speed [pr] (#6880)	2024-10-04 18:14:23 +08:00
models	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
testextra	Hook memoryview via class instead of a function (#7627)	2024-11-11 09:07:06 +08:00
unit	Context manager support for tqdm (#7770)	2024-11-18 14:12:03 -05:00
__init__.py	All devices are equal! (#196)	2020-12-15 23:44:08 -08:00
Dockerfile	new cloud is cloudy [pr] (#7631)	2024-11-11 20:18:04 +08:00
helpers.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_arange.py	default threefry (#6116)	2024-09-25 17:45:13 +08:00
test_assign.py	Add replicate mode to Tensor.pad (#7608)	2024-11-18 10:55:38 -05:00
test_compile_failures.py	enable a few tests that work now (#7721)	2024-11-15 14:30:52 -05:00
test_const_folding.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_conv.py	db in wal mode (#5388)	2024-07-12 20:43:36 -07:00
test_conv_shapetracker.py	s/UOps/Ops (#7500)	2024-11-03 11:26:10 +08:00
test_copy_speed.py	node cleanup + local metal test speed [pr] (#6880)	2024-10-04 18:14:23 +08:00
test_device_speed.py	remove iter from uopgraph (#6110)	2024-08-16 15:58:29 -07:00
test_dtype.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_dtype_alu.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_fusion_op.py	Ops.VALID cleanup, move recursive tests [pr] (#7713)	2024-11-15 20:22:46 +08:00
test_fuzz_shape_ops.py	faster ci [pr] (#7348)	2024-10-29 14:01:44 +08:00
test_gc.py	seed in tensor (#6869)	2024-10-06 14:46:58 -04:00
test_graph.py	mask out writable bufs in runtime access_resources (#7234)	2024-10-23 16:13:50 +03:00
test_hcq.py	vcount + cleanups (#7393)	2024-10-30 12:50:15 +08:00
test_image_dtype.py	image dtype fusion tests [pr] (#7530)	2024-11-04 22:00:16 +08:00
test_jit.py	skip test_jit_batch_split if JIT >= 2 (#7561)	2024-11-05 14:59:04 -05:00
test_kernel_cache.py	move the compiler cache to be global (#2957)	2024-01-01 10:59:56 -08:00
test_lazybuffer.py	to_uop one path for all ops part 1 (#7745)	2024-11-17 20:12:44 +08:00
test_linearizer.py	Add replicate mode to Tensor.pad (#7608)	2024-11-18 10:55:38 -05:00
test_linearizer_dumb.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_linearizer_failures.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_linearizer_overflows.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_masked_st.py	multitensor start (#2676)	2023-12-07 17:07:05 -08:00
test_metal.py	METAL raise RuntimeError with no compiler and bad src (#7603)	2024-11-08 17:09:12 -05:00
test_method_cache.py	simple LoadOps.ASSIGN (#3745)	2024-03-14 20:44:34 -07:00
test_multitensor.py	fix Tensor(MultiLazyBuffer) with different dtype should fail (#7757)	2024-11-17 21:05:45 -05:00
test_net_speed.py	nv mockgpu (#4600)	2024-05-15 23:46:08 +03:00
test_nn.py	load_state_dict test cases when model and data shard differently (#7774)	2024-11-18 16:08:24 -05:00
test_ocl.py	Fix GPT2 with OpenCL backend. (#6821)	2024-10-01 16:57:22 +08:00
test_ops.py	Add replicate mode to Tensor.pad (#7608)	2024-11-18 10:55:38 -05:00
test_optim.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_pickle.py	Revert "s/UPat/Pat (#7506)" [pr] (#7517)	2024-11-03 16:33:02 -05:00
test_profiler.py	remove pyint [pr] (#7016)	2024-10-12 22:36:24 +08:00
test_randomness.py	add tiny test for randomness + remove ulong buffers (#7648)	2024-11-12 12:45:52 +08:00
test_rearrange_einops.py	Missing features from rearrange (#6184)	2024-08-19 11:19:07 -07:00
test_renderer_failures.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_sample.py	UOp.variable (#7010)	2024-10-12 18:20:44 +08:00
test_schedule.py	Add replicate mode to Tensor.pad (#7608)	2024-11-18 10:55:38 -05:00
test_search.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_setitem.py	disable test_setitem_inplace_operator again (#7495)	2024-11-02 19:01:23 -04:00
test_specific_conv.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_speed_v_torch.py	fix HALF=1 in test_speed_v_torch (#7642)	2024-11-11 14:29:37 -05:00
test_subbuffer.py	move lazy to engine [pr] (#6886)	2024-10-04 23:19:26 +08:00
test_symbolic_jit.py	UOp.variable (#7010)	2024-10-12 18:20:44 +08:00
test_symbolic_ops.py	UOp.variable (#7010)	2024-10-12 18:20:44 +08:00
test_symbolic_shapetracker.py	fix view add with symbolic shape (#7569)	2024-11-06 11:39:20 -05:00
test_tensor.py	simpler rand_like (#7680)	2024-11-13 12:28:41 -05:00
test_tensor_data.py	BEAM_COMPARE=2 validates the correctness of BEAM kernels (#5458)	2024-07-13 13:53:43 -07:00
test_tensor_variable.py	combine pad2d with pad (#7677)	2024-11-14 17:56:02 +08:00
test_tiny.py	add tiny test for randomness + remove ulong buffers (#7648)	2024-11-12 12:45:52 +08:00
test_to_numpy.py	Apply ruff linting rules to tests (#2473)	2023-11-27 21:24:06 -08:00
test_transcendental.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_uop_graph.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_uops.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_uops_stats.py	update tests (#7554)	2024-11-05 11:35:15 -05:00
test_viz.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_winograd.py	s/UOps/Ops (#7500)	2024-11-03 11:26:10 +08:00
test_zero_copy.py	remove numpy from device (#3123)	2024-01-14 19:36:05 -08:00