Clément Verrier 0e409ff5ce fix indentation in UOp pretty_print for repeated references (#13857) ...

* fix correct indentation in UOp pretty_print for repeated references

When a UOp was referenced multiple times, the walrus operator notation
(e.g., x0:=) was correctly used for the first occurrence, but subsequent
references had misaligned indentation due to an extra space character.

Fix indentation misalignment in pretty_print() when UOps are referenced
multiple times.

* add simple unit tests for UOp repr

---------

Co-authored-by: chenyu <chenyu@fastmail.com>

2025-12-29 10:46:16 -05:00

..

device

mockgpu: async support (#13868)

2025-12-29 13:18:37 +03:00

external

add optional compiler in Renderer (#13817)

2025-12-23 17:58:46 -05:00

mockgpu

write python emulator from RDNA3 psuedocode in pdf (#13841)

2025-12-29 07:39:53 -05:00

models

mark slow tests as slow instead of as CI (#13736)

2025-12-17 10:29:57 -04:00

opt

Revert "rewrites for renderer and compiler (#13646)" (#13806)

2025-12-22 19:21:33 -05:00

speed

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

testextra

use new style amd compiler in viz (#13848)

2025-12-27 23:59:30 +09:00

unit

fix indentation in UOp pretty_print for repeated references (#13857)

2025-12-29 10:46:16 -05:00

web

remove stale / untested viz related files (#13785)

2025-12-21 16:42:48 +08: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

no full_rewrite [pr] (#13809)

2025-12-22 23:20:01 -05:00

test_arange.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_assign.py

skip test_double_assign in CI LVP (#13620)

2025-12-08 14:54:02 -05:00

test_compile_failures.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04: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_alu.py

does this fix the dtype test? (#13779)

2025-12-20 17:31:46 -04:00

test_dtype.py

Handle missing bfloat16 natives on CPU architectures (#13553)

2025-12-11 15:38:43 -05:00

test_edgecases.py

Mesa NIR backend (NAK/LLVMpipe) (#12089)

2025-10-15 17:38:33 +08:00

test_fusion_op.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_gc.py

fix a few tests [pr] (#13498)

2025-11-29 13:43:45 -08:00

test_graph.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_image_dtype.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04: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_footguns.py

add test for jit footguns (#13701)

2025-12-15 10:47:44 -05:00

test_jit.py

failed test case when init jit with empty inputs (#13641)

2025-12-10 22:03:06 -05:00

test_kernel_cache.py

make method_cache account for compiler (#12156)

2025-09-13 17:00:11 +03: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_linearizer.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04: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

Revert "optim empty shard #13513 (#13598)" (#13855)

2025-12-27 21:10:23 -05:00

test_nn.py

mark slow tests as slow instead of as CI (#13736)

2025-12-17 10:29:57 -04:00

test_ops.py

mark slow tests as slow instead of as CI (#13736)

2025-12-17 10:29:57 -04:00

test_opt_gemm.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_optim.py

mark slow tests as slow instead of as CI (#13736)

2025-12-17 10:29:57 -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_range.py

remove dtype from range, it will be dtypes.index soon [pr] (#11914)

2025-08-29 09:52:07 -07:00

test_outerworld.py

vmap on full model (#13340)

2025-11-18 16:06:06 -08: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

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_randomness.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_rangeify.py

mark slow tests as slow instead of as CI (#13736)

2025-12-17 10:29:57 -04:00

test_renderer_failures.py

no full_rewrite [pr] (#13809)

2025-12-22 23:20:01 -05:00

test_schedule.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_setitem.py

remove UOp.st (#12716)

2025-10-16 14:44:09 +08:00

test_softmax_fusion.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04: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_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 (#13718)

2025-12-16 13:55:06 -04:00

test_tensor.py

no full_rewrite [pr] (#13809)

2025-12-22 23:20:01 -05: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_stats.py

no full_rewrite [pr] (#13809)

2025-12-22 23:20:01 -05:00

test_uops.py

handle empty VECTORIZE in UOp.render() (#13847)

2025-12-27 10:09:39 -05:00

test_zero_copy.py

unify cpu and llvm (#11982)

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