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_alu.py

python float8 support (#11960)

2025-09-18 09:17:09 -04:00

test_dtype.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_cases.py

enumerate cases of Tensors in the JIT (#10548)

2025-05-28 11:51:27 -07:00

test_jit.py

failing rng test (#12328)

2025-09-29 16:06:45 +08:00

test_kernel_cache.py

make method_cache account for compiler (#12156)

2025-09-13 17:00:11 +03:00

test_linearizer_dumb.py

Lower index dtype spec fix (#12337)

2025-09-30 04:26:50 +02:00

test_linearizer.py

fix some rangeify tests (#12370)

2025-09-30 20:12:08 +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

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_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_tensor.py

RANGEIFY test_tensor (#12235)

2025-09-18 10:35:43 -04: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_stats.py

fix some rangeify tests (#12370)

2025-09-30 20:12:08 +08:00

test_uops.py

remove TestShapeSpec, it relies on ShapeTracker [pr] (#12369)

2025-09-30 14:20:35 +03:00

test_zero_copy.py

unify cpu and llvm (#11982)

2025-09-09 13:54:44 +03:00