hooved 969a1b35ca LR scheduler for Stable Diffusion mlperf training (#12201) ... * add lr scheduler for stable diffusion training * add lr scheduler test * rerun ci * rerun CI * use np for testing * move test to CI path * remove unneeded copy		2025-09-30 21:21:08 -04:00
..
device	Support bfloat16 on NULL backend (#12340)	2025-09-30 00:02:30 -04:00
external	LR scheduler for Stable Diffusion mlperf training (#12201)	2025-09-30 21:21:08 -04:00
mockgpu	mockgpu: no hang if gpuocelot failed (#11915)	2025-08-30 00:44:49 +03:00
models	fix cifar training in RANGEIFY (#12355)	2025-09-30 15:59:19 +08:00
opt	Check for group inside another reduce (#12268)	2025-09-23 00:32:41 +02: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	update Tensor unit tests for RANGEIFY (#12359)	2025-09-30 11:17:21 +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	add arange cat arange test (#12217)	2025-09-16 17:12:32 -04:00
test_assign.py	rangeify permuted assign (#12299)	2025-09-29 07:27:57 +03:00
test_compile_failures.py	unify cpu and llvm (#11982)	2025-09-09 13:54:44 +03:00
test_const_folding.py	rangeify: only test correctness in multi (#12339)	2025-09-30 09:55:59 +03:00
test_dtype.py	python float8 support (#11960)	2025-09-18 09:17:09 -04:00
test_dtype_alu.py	python float8 support (#11960)	2025-09-18 09:17:09 -04: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	Rangeify IMAGE (#12304)	2025-09-26 07:21:02 +02:00
test_interop.py	hotfix: interop example (#9237)	2025-02-25 10:32:00 +03:00
test_jit.py	failing rng test (#12328)	2025-09-29 16:06:45 +08: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	fix some rangeify tests (#12370)	2025-09-30 20:12:08 +08:00
test_linearizer_dumb.py	Lower index dtype spec fix (#12337)	2025-09-30 04:26:50 +02: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	rangeify: only test correctness in multi (#12339)	2025-09-30 09:55:59 +03:00
test_nn.py	test rangeify const folding (#12200)	2025-09-15 20:03:48 -04:00
test_ops.py	remu: add new instructions introduced in RANGEIFY (#12363)	2025-09-30 12:36:29 +03: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	rangeify: update test_pickle asserts (#12366)	2025-09-30 13:27:41 +03: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	dsp opt test has different axis number on rangeify (#12309)	2025-09-27 05:06:11 +02:00
test_randomness.py	move test_sample to test_randomness (#12266)	2025-09-21 21:11:32 -04:00
test_rangeify.py	bad conv in rangeify (#12373)	2025-10-01 08:56:22 +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_schedule.py	rangeify: buf limit (#12336)	2025-09-30 14:59:47 +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	fix some rangeify tests (#12370)	2025-09-30 20:12:08 +08: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	Shrink instead of reshape to unregister symbolic (#12241)	2025-09-19 06:04:35 +02:00
test_symbolic_ops.py	Shrink instead of reshape to unregister symbolic (#12241)	2025-09-19 06:04:35 +02:00
test_tensor.py	RANGEIFY test_tensor (#12235)	2025-09-18 10:35:43 -04:00
test_tensor_uop.py	rename lazydata to uop (#10698)	2025-06-08 08:42:22 -07:00
test_tensor_variable.py	Shrink instead of reshape to unregister symbolic (#12241)	2025-09-19 06:04:35 +02:00
test_tiny.py	Shrink instead of reshape to unregister symbolic (#12241)	2025-09-19 06:04:35 +02: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	z3 fix loaded mask (#12353)	2025-09-30 06:55:50 +02:00
test_uops.py	remove TestShapeSpec, it relies on ShapeTracker [pr] (#12369)	2025-09-30 14:20:35 +03:00
test_uops_stats.py	fix some rangeify tests (#12370)	2025-09-30 20:12:08 +08:00
test_zero_copy.py	unify cpu and llvm (#11982)	2025-09-09 13:54:44 +03:00