George Hotz 3fafc4d670 support symbolic shapes in split/chunk when split dim is concrete ... Previously split() and chunk() required all dimensions to be concrete. Now they only require the dimension being split to be concrete, allowing them to work with tensors that have symbolic shapes in other dimensions. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>		2025-12-16 12:28:24 -04:00
..
device	fix a few tests [pr] (#13498)	2025-11-29 13:43:45 -08:00
external	am: support xgmi systems (#13659)	2025-12-12 18:55:45 +03:00
mockgpu	nv: minimal hevc (#13502)	2025-11-30 16:46:55 +03:00
models	add Swish op (#13541)	2025-12-08 12:41:18 -05:00
opt	require renderer argument in get_program, removes device opening in process replay [pr] (#13524)	2025-12-03 02:05:31 +08:00
speed	ops_gpu -> ops_cl (#12103)	2025-09-10 15:15:48 -04:00
testextra	tk: kernel finish groups stores (#13704)	2025-12-15 09:16:17 -08:00
unit	qwen model is working (#13690)	2025-12-15 18:00:34 -04:00
web	time viz (#10763)	2025-06-17 19:39:34 +03: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	add needs_second_gpu decorator (#13543)	2025-12-02 19:08:23 -08:00
test_arange.py	require renderer argument in get_program, removes device opening in process replay [pr] (#13524)	2025-12-03 02:05:31 +08:00
test_assign.py	skip test_double_assign in CI LVP (#13620)	2025-12-08 14:54:02 -05:00
test_compile_failures.py	Mesa NIR backend (NAK/LLVMpipe) (#12089)	2025-10-15 17:38:33 +08:00
test_const_folding.py	shared_codegen_spec and fix index spec (#12967)	2025-10-29 09:14:11 +01:00
test_custom_kernel.py	multi custom kernel support (#13716)	2025-12-16 11:36:30 -04:00
test_dtype.py	Handle missing bfloat16 natives on CPU architectures (#13553)	2025-12-11 15:38:43 -05:00
test_dtype_alu.py	add loads at the end (#12988)	2025-10-30 10:42:19 +08:00
test_edgecases.py	Mesa NIR backend (NAK/LLVMpipe) (#12089)	2025-10-15 17:38:33 +08:00
test_fusion_op.py	Cleanup const buffers (#12829)	2025-10-21 14:53:49 +02:00
test_gc.py	fix a few tests [pr] (#13498)	2025-11-29 13:43:45 -08:00
test_graph.py	add needs_second_gpu decorator (#13543)	2025-12-02 19:08:23 -08:00
test_image_dtype.py	Cleanup const buffers (#12829)	2025-10-21 14:53:49 +02:00
test_interop.py	hotfix: interop example (#9237)	2025-02-25 10:32:00 +03:00
test_jit.py	failed test case when init jit with empty inputs (#13641)	2025-12-10 22:03:06 -05:00
test_jit_cases.py	enumerate cases of Tensors in the JIT (#10548)	2025-05-28 11:51:27 -07:00
test_jit_footguns.py	add test for jit footguns (#13701)	2025-12-15 10:47:44 -05:00
test_kernel_cache.py	make method_cache account for compiler (#12156)	2025-09-13 17:00:11 +03:00
test_linearizer.py	fix cifar while keeping openpilot fused (#13528)	2025-12-02 12:05:56 -08:00
test_linearizer_dumb.py	add loads at the end (#12988)	2025-10-30 10:42:19 +08:00
test_linearizer_failures.py	require renderer argument in get_program, removes device opening in process replay [pr] (#13524)	2025-12-03 02:05:31 +08:00
test_memory_planner.py	memplan: fix large buffers planning (#11878)	2025-08-27 23:54:27 +03:00
test_method_cache.py	make method_cache account for compiler (#12156)	2025-09-13 17:00:11 +03:00
test_multitensor.py	optim empty shard #13513 (#13598)	2025-12-09 12:28:36 -05:00
test_nn.py	fix and test loading num_batches_tracked (#13538)	2025-12-04 01:22:49 -08:00
test_ops.py	test: enable cummax scalar IndexError test (#13625)	2025-12-09 12:25:56 -05:00
test_opt_gemm.py	require renderer argument in get_program, removes device opening in process replay [pr] (#13524)	2025-12-03 02:05:31 +08:00
test_optim.py	use torch 2.9 and its Muon in test (#12773)	2025-10-21 13:35:17 -04:00
test_opts.py	require renderer argument in get_program, removes device opening in process replay [pr] (#13524)	2025-12-03 02:05:31 +08:00
test_outerworld.py	vmap on full model (#13340)	2025-11-18 16:06:06 -08:00
test_outerworld_range.py	remove dtype from range, it will be dtypes.index soon [pr] (#11914)	2025-08-29 09:52:07 -07:00
test_pickle.py	no sym in pm_reduce [pr] (#13398)	2025-11-20 16:49:09 -05:00
test_profiler.py	test_graph is flaky	2025-11-26 16:37:42 -08:00
test_quantize_onnx.py	require renderer argument in get_program, removes device opening in process replay [pr] (#13524)	2025-12-03 02:05:31 +08:00
test_randomness.py	add needs_second_gpu decorator (#13543)	2025-12-02 19:08:23 -08:00
test_rangeify.py	fix indexing bug with convs	2025-11-07 16:45:19 -08:00
test_remote.py	skip more multi remote tests (#12410)	2025-10-02 04:50:46 +03:00
test_renderer_failures.py	add loads at the end (#12988)	2025-10-30 10:42:19 +08:00
test_schedule.py	add SGD to beautiful_mnist (#13571)	2025-12-04 12:17:29 -05:00
test_setitem.py	remove UOp.st (#12716)	2025-10-16 14:44:09 +08:00
test_softmax_fusion.py	fuse hasn't worked since rangeify, remove it (#13057)	2025-11-02 14:01:52 +08:00
test_stunning.py	remove FUSE_ARANGE (#12511)	2025-10-08 04:54:07 -04:00
test_subbuffer.py	add needs_second_gpu decorator (#13543)	2025-12-02 19:08:23 -08:00
test_symbolic_jit.py	remove trivial use of RANGEIFY flag (#12550)	2025-10-09 02:29:38 -04:00
test_symbolic_ops.py	update test_conv2d_ceildiv_edge_case (#12779)	2025-10-18 16:43:32 -04:00
test_tensor.py	use assign in Tensor.backward (#13674)	2025-12-13 22:43:06 -05:00
test_tensor_uop.py	new linearizer with early endrange (#12823)	2025-10-21 17:37:48 +08:00
test_tensor_variable.py	support symbolic shapes in split/chunk when split dim is concrete	2025-12-16 12:28:24 -04:00
test_tiny.py	beautiful SQTT_PARSE=1 with color (#13428)	2025-11-23 01:05:14 -08:00
test_to_numpy.py	Apply ruff linting rules to tests (#2473)	2023-11-27 21:24:06 -08:00
test_transcendental.py	Mesa NIR backend (NAK/LLVMpipe) (#12089)	2025-10-15 17:38:33 +08:00
test_uop_graph.py	fix after end (#13542)	2025-12-02 18:42:58 -08:00
test_uops.py	Update tests (#13479)	2025-11-28 18:35:28 -08:00
test_uops_stats.py	require renderer argument in get_program, removes device opening in process replay [pr] (#13524)	2025-12-03 02:05:31 +08:00
test_zero_copy.py	unify cpu and llvm (#11982)	2025-09-09 13:54:44 +03:00