Shun Usami 2a72b00679 Add test for 2D tensor indexing in setitem (#12193) ... * Add test for 2D tensor indexing in setitem * Fix _masked_setitem to handle multi dim indexing correctly * Fix indent * Add fuzz test for 3D tensor indexing in setitem * Skip indexing fuzz test (slow)		2025-09-16 14:57:25 -04:00
..
device	ops_gpu -> ops_cl (#12103)	2025-09-10 15:15:48 -04:00
external	make method_cache account for compiler (#12156)	2025-09-13 17:00:11 +03:00
mockgpu	mockgpu: no hang if gpuocelot failed (#11915)	2025-08-30 00:44:49 +03:00
models	update test/models (#12142)	2025-09-12 16:43:28 -04:00
opt	ops_gpu -> ops_cl (#12103)	2025-09-10 15:15:48 -04:00
speed	ops_gpu -> ops_cl (#12103)	2025-09-10 15:15:48 -04:00
testextra	break swizzle into three chunks [pr] (#11153)	2025-07-09 15:30:34 -07:00
unit	remove PROFILE=1 option, it's just VIZ=1 [pr] (#12176)	2025-09-15 12:51:50 +03: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	ops_gpu -> ops_cl (#12103)	2025-09-10 15:15:48 -04:00
test_arange.py	early reduce simplify (#12046)	2025-09-10 21:02:46 +08:00
test_assign.py	rename lazydata to uop (#10698)	2025-06-08 08:42:22 -07:00
test_compile_failures.py	unify cpu and llvm (#11982)	2025-09-09 13:54:44 +03:00
test_const_folding.py	clean up TestBitcastConstFolding (#11856)	2025-08-26 15:26:47 -04:00
test_dtype.py	remove ml_dtypes (#12169)	2025-09-14 14:20:05 -04:00
test_dtype_alu.py	do not use getenv('PTX') in tests (#12095)	2025-09-10 14:04:07 +03:00
test_edgecases.py	add dtypes.index (#12015)	2025-09-06 06:03:44 +02:00
test_fusion_op.py	just schedule in test_recursive_pad [pr] (#8860)	2025-02-02 15:01:24 +02:00
test_gc.py	s/lb_refcount/uop_refcount [pr] (#10865)	2025-06-18 21:48:04 +03:00
test_graph.py	hcq: cpu can be graphed (#11474)	2025-08-02 21:01:19 +03:00
test_image_dtype.py	ops_gpu -> ops_cl (#12103)	2025-09-10 15:15:48 -04:00
test_interop.py	hotfix: interop example (#9237)	2025-02-25 10:32:00 +03:00
test_jit.py	unify cpu and llvm (#11982)	2025-09-09 13:54:44 +03:00
test_jit_cases.py	enumerate cases of Tensors in the JIT (#10548)	2025-05-28 11:51:27 -07:00
test_kernel_cache.py	make method_cache account for compiler (#12156)	2025-09-13 17:00:11 +03:00
test_linearizer.py	update outbufs selection in test_linearizer [pr] (#12166)	2025-09-14 13:46:49 +03:00
test_linearizer_dumb.py	early reduce simplify (#12046)	2025-09-10 21:02:46 +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	smaller test_simple_reduce (#12124)	2025-09-11 15:45:38 -04:00
test_nn.py	test rangeify const folding (#12200)	2025-09-15 20:03:48 -04:00
test_ops.py	update torch 2.8 (#12172)	2025-09-14 15:19:03 -04:00
test_opt_gemm.py	delete more tests (#12043)	2025-09-05 15:31:30 -07:00
test_optim.py	[bounty] Muon optim (#11414)	2025-08-13 14:27:55 -04:00
test_opts.py	ops_gpu -> ops_cl (#12103)	2025-09-10 15:15:48 -04: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	move into codegen late [pr] (#11823)	2025-08-24 10:23:25 -07:00
test_profiler.py	remove PROFILE=1 option, it's just VIZ=1 [pr] (#12176)	2025-09-15 12:51:50 +03:00
test_quantize_onnx.py	delete more tests (#12043)	2025-09-05 15:31:30 -07:00
test_randomness.py	do not use getenv('PTX') in tests (#12095)	2025-09-10 14:04:07 +03:00
test_rangeify.py	use tags instead of graph_rewrite_map in rangeify (#12110)	2025-09-14 11:39:01 +08:00
test_remote.py	Remote scheduler changes (#11177)	2025-07-21 09:29:44 -07:00
test_renderer_failures.py	Move Ops.SPECIAL arg to src (#11918)	2025-09-04 09:31:44 +02:00
test_sample.py	enable WEBGPU tests with buffer limit (#11489)	2025-08-03 13:02:44 -07:00
test_schedule.py	assert kernel count (#12205)	2025-09-16 14:24:39 +03:00
test_setitem.py	Add test for 2D tensor indexing in setitem (#12193)	2025-09-16 14:57:25 -04:00
test_softmax_fusion.py	a*(1/b) -> a/b on LLVM, CPU (#11743)	2025-08-20 09:35:10 -04:00
test_stunning.py	move bind to big graph (#11539)	2025-08-06 13:27:51 -07:00
test_subbuffer.py	redundant code (#11014)	2025-06-29 09:06:10 -07:00
test_symbolic_jit.py	failed test case for symbolic pad (#12179)	2025-09-15 00:25:21 -04:00
test_symbolic_ops.py	failed test case for symbolic pad (#12179)	2025-09-15 00:25:21 -04:00
test_tensor.py	Tensor.pad_to and Tensor.shrink_to (#12210)	2025-09-16 12:24:55 -04:00
test_tensor_uop.py	rename lazydata to uop (#10698)	2025-06-08 08:42:22 -07:00
test_tensor_variable.py	[bounty] Remove using reshape to register symbolic shape (#11771)	2025-08-28 12:30:49 -04:00
test_tiny.py	isolate the const hacks in the old kernelize (#12126)	2025-09-12 08:35:35 +08:00
test_to_numpy.py	Apply ruff linting rules to tests (#2473)	2023-11-27 21:24:06 -08:00
test_transcendental.py	Fix DSP transcendentals (#9542)	2025-03-22 11:08:18 +08:00
test_uop_graph.py	update store load noop pattern to use Invalid (#12141)	2025-09-12 22:25:53 +02:00
test_uops.py	ops_gpu -> ops_cl (#12103)	2025-09-10 15:15:48 -04:00
test_uops_stats.py	do not use getenv('PTX') in tests (#12095)	2025-09-10 14:04:07 +03:00
test_zero_copy.py	unify cpu and llvm (#11982)	2025-09-09 13:54:44 +03:00