Commit Graph

Select branches

Hide Pull Requests

abstractions_4

add_local_buffer

after_qol

algebraic_upat

all_bv_no_idx

all_sink

amd_isel

amd_refac

amd_smu_reset_patch

amd_sqtt

amd_uop

amd_uop_cleanups

assign_copy

assign_try_3

before2

better_spec

better_syn

bitcast_spec

broadcast_shape_2

bt_ei

buf_heur

bump_nv_2404

cache_reshape

call_inline

call_is_kernels

callify

cast_vec_as_cast

cat_mop

changes_from_rangify

check_clsetarg

check_dims

cifar_30

cifar_30_clean

clean_load

clean_spec

clone_tg

codegen_try_2

contig_on_after

continue_mat

deepsek

default_late_allreduce

delete_kernel_py

delete_ones

delete_slow_rangeify

delta_attention

delta_attention_scan

demote_op

disable_rules

dsp_search

dsp_search_merged

dtype_shape

earlier_gater

emu_cdna

emulate_ctx

end_all_ranges

endrange

er_prioity

execution_unit

failing_assign

fancy_usb

fast_llm

fix_fa_fwd_clang_22

fix_fuse

fix_llvm_cast

fix_range_merging_sd

fix_rangeify_tests

fix_rc

fix_sched_copy

fix_slice_store

fix_src_spec

fixup_fix_kernel

full_spec

fun_w_egraphs

gate-on-load-store

gate_rewrite

gemma4_gpt

gen_pdf_fast

gh-pages

gl_dims

gldims

glm_flash

grad_outerworld

grok_x86_novec

group_div_rules

hcq2_vram_copies

heur_4096

heur_group

hk_gemm

http_threading

hwtest_fixes

id_store

if_range_cleanups

ignore_oob

image_idx_xy

image_in_mc

image_no_vec

image_refactors_stack

index_to_slice

invalid_try_3

jinja_template

kernel4

kernel_is_call

kernelless

kimi_slop

kitten_matmul

late_Locals

late_add_load

late_allreduce

late_codegen

late_decomp_is_late

late_lower_idx

late_view

late_weakint

lds_double_buffering

less_wmma_arg

lil_dg_change

lil_rangeify_td

llama_trainer

llm_262k_minimal

llm_flash_attn

llm_qwen

llm_serve_fixes

load_view_late

local_cache

locals_have_device

loop_end_op

loop_is_rng

lowerer_hacks

mac_pytest

manual_kernel_3

master

mega_lowerer

min_new_lin

min_outer_range

minor_views

mixins7

mlperf_training_v4.1

mlperf_training_v5.0

mlperf_training_v6.0

modern_scan

more_apps

more_early_comps

more_folding_rules

more_kernelcount

more_pcode

more_rangeify_pm_tests

more_ren_clean

more_uop_gc_test

more_uop_mm

more_vmap

move_devec

move_dtype_decomps

move_expander

move_gates_to_load_store

move_mop_cleanup

move_schedule

move_simp_views

moveleftright

mt_typing

multi_range

multioutput

new_ck

new_devec

new_invalid_and_image

new_linearizer

new_shape

new_test

no_assign_2

no_buf_0

no_decimals

no_late_simplify_marg

no_merge_ends

no_merge_views

no_mstack_in_allreduce

no_one_ranges

no_sip_usbgpu

no_tsink_base

no_tuplize

no_user_ck_contig

no_vec_x86_4

no_weakint_variable

noop_index

novecimg

num_batches_tracked_shape

olmo3_support

one_compiler

one_hcopt

only_assign_buffers

only_loop_backends

only_reg_emu

only_reg_emu2

outer_range

outerworld_work

pad_invalid_work

pad_mixin

param_call_is_function

parse_sqtt_more

pci_iommu

port_nir

postrange_hcopts

precompile_backward

prefill_qwen_fast

print_special_ops

ps_clean

pyremu_try

python_prg

python_speed

q4k_support

qcom-openpilot-13ms

qcom_830

qcom_mmapeak

qcom_speed_codex

qol_string

qwen36_27b_amd_900

qwen36_27b_amd_minimal

qwen36_amd_bugfixes

qwen_mergable

r_cleanups

r_new_opt

ranged_call

rangeify_2

rangeify_const_folding

rdna3_vibes

rdna4_asm

rdna4_gemm

real_pyrender

recurse_sub

reg_coalese

relu_pattern

remove_assign

remove_cast_hack

remove_ck_contig

remove_define_var

remove_forced_re

remove_on_stack

remove_programspec

remove_pylint

remove_slow_tests

remove_vec_2

remove_vec_3

renderers_use_shapes_2

replace_if_with_range

reprocess_node_experiments

reshape_trait

resnet-ci

revert-11547-fix_mismatch_red

revert-11715-revert-11711-amd_llvm_tc_cast

revert-11809-no_ki_in_gpudims

revert-11811-better_div_nesting

revert-11876-memplan_vbig

revert-12104-simplify_valid_in_symbolic

revert-12308-bufferize_early

revert-12731-jitlink_init_check

revert-12813-after_cleanups

revert-13046-better_cleanup_arange_buffers

revert-13311-prof_link_sqtt

revert-13583-rocr_desc_2

revert-14734-tload

revert-17141-full_rewrite_to_sink

revert-17408-_broadcasted-weak

revert-9438-simple_matmul_np_rand

revert_wrong_ck_fix

rewrite_rangeify2

rtoposort

scan_assign

scan_op_8

sched_cache

sched_lin

sd_load_simple

self_type

set_testpath

shrink_in_render

simp_spec

simple_gidxs

simple_pool

simple_priority

simple_rangeify

simpler-sd-compile

simpler_end

simpler_fusion

simpler_image

simpler_postrange

simpler_reduce

slice_like_shrink

spec_3

spec_cleanup

spec_ren

speed_one_linear

split_buf_idx

split_decomp_pass

split_sub

stack_0

store_is_void

string_ranges

support_opts_in_contig

swizzle_in_kernel

sym_llm

sym_work

symbolic-chunk-split

szdiff_bot_test

tc_is_heur

tensor_graph_single_sink

test_amd_crash

test_fa

test_from_postopt

test_hgemm

test_unit

tiny_scan

tmr

tool_call_support

topoprop

ttomsa-new_linearizer

tuplegtuple

two_stage_remove

typed_checks

typing_work

unused_permute_arg

uop_prg

upd_test4

update_benchmark

update_benchmark_search

update_mlperf

user_failure

vecless_x86_glm

verify_sink_dims_clean

vf_ish

view_after_const

view_in_codegen

viz_cbroadcast

warp_fun

wgpu-f16-packed

working_postopt

x86_moves

x86_numel

v0.10.0

v0.10.1

v0.10.2

v0.10.3

v0.11.0

v0.12.0

v0.13.0

v0.4.0

v0.5.0

v0.6.0

v0.7.0

v0.8.0

v0.9.0

v0.9.1

v0.9.2

Mono Color

• ea7672931f fix test_matmul_relu_cat (#12478) George HotzandGitHub 2025-10-07 14:32:23 +08:00
• 2650f15c4b fix test_matmul_relu_cat fix_rc geohot 2025-10-07 14:15:31 +08:00
• 3fae886aa9 Merge branch 'master' into clone_tg George HotzandGitHub 2025-10-07 14:02:36 +08:00
• 514d2a0774 merge tagless reshapes (#12474) George HotzandGitHub 2025-10-07 13:57:58 +08:00
• 7b48f3cc45 failed test case repro for openpilot model (#12475) chenyuandGitHub 2025-10-07 13:46:43 +08:00
• 3f44ef699f Merge branch 'master' into clone_tg George HotzandGitHub 2025-10-07 13:08:35 +08:00
• fa23f37e33 clone thunderkittens in uops geohot 2025-10-07 13:06:58 +08:00
• 284db26a12 cleanup geohot 2025-10-07 12:13:01 +08:00
• 0a0cb0b9e8 merge tagless reshapes geohot 2025-10-07 12:11:11 +08:00
• a5484b767e remove skipping cast in simplify_valid [pr] (#12472) chenyuandGitHub 2025-10-07 12:10:04 +08:00
• b4509fba31 thundermittens (#12471) George HotzandGitHub 2025-10-07 11:47:39 +08:00
• 0f25b4b289 move frontend dir to nn [pr] (#12470) George HotzandGitHub 2025-10-07 10:42:22 +08:00
• f664bcc8bd use recursive_property in UOp tracing (#12469) qazalandGitHub 2025-10-06 21:10:52 +03:00
• e4ec4d2c51 remove noops add_local_buffer geohot 2025-10-06 23:21:48 +08:00
• 1d0b114a7b flip that geohot 2025-10-06 19:25:05 +08:00
• 51301c3b22 no locals geohot 2025-10-06 19:18:26 +08:00
• 17644fc304 gate pipeline geohot 2025-10-06 19:02:21 +08:00
• bf59379741 pipeline works on tc geohot 2025-10-06 18:59:51 +08:00
• 1af05dae77 fix rangeify in compile4.py (#12467) qazalandGitHub 2025-10-06 13:37:46 +03:00
• 97f122b591 tensor core works geohot 2025-10-06 18:34:03 +08:00
• afe31cc92a it works geohot 2025-10-06 17:54:05 +08:00
• 76e8a3250c rangeify: late zero folding (#12464) qazalandGitHub 2025-10-06 12:52:33 +03:00
• 3444e414f6 Merge branch 'master' into add_local_buffer George HotzandGitHub 2025-10-06 16:17:08 +08:00
• 0c015a24fe use recursive_property to prevent RecursionError (#12465) George HotzandGitHub 2025-10-06 15:59:18 +08:00
• a1881b0c17 update test_chicken (#12466) chenyuandGitHub 2025-10-06 15:58:44 +08:00
• 39d8459ff2 comments geohot 2025-10-06 14:13:31 +08:00
• fdc0489e18 pipelining wip geohot 2025-10-06 12:31:20 +08:00
• 1b1978b9c0 early copy fixup (#12463) qazalandGitHub 2025-10-06 06:38:29 +03:00
• c1e85f699c multi test case for sharded ring allreduce (#12462) chenyuandGitHub 2025-10-06 11:18:24 +08:00
• 1823a5043f don't check MAX_BUFFER_SIZE on NULL (#12461) chenyuandGitHub 2025-10-06 10:09:29 +08:00
• 46e8ea15c1 split pm_substitute_recurse (#12460) George HotzandGitHub 2025-10-06 09:35:50 +08:00
• 061a17ee4d split pm_substitute_recurse split_sub geohot 2025-10-06 09:22:01 +08:00
• df1b379a36 Merge branch 'master' into add_local_buffer George HotzandGitHub 2025-10-06 08:58:46 +08:00
• 5a9acee9b7 fixes ttomsa 2025-10-05 22:07:09 +01:00
• 1ffff0a801 draft ttomsa 2025-10-05 20:22:50 +01:00
• 1216fff781 remote: raise runtimeerror in checkz (#12453) nimlgenandGitHub 2025-10-05 21:22:53 +08:00
• 6ad9a688ed add failing test after "pend substitutes for speed" (#12457) qazalandGitHub 2025-10-05 16:10:04 +03:00
• 74b04f7dca test beautiful_mnist_multigpu (#12455) chenyuandGitHub 2025-10-05 20:45:01 +08:00
• 69857d0ab0 Stable Diffusion mlperf training (#11304) 3 people 2025-10-05 07:56:05 -04:00
• a976ace404 minor improvements to rewrite (#12454) George HotzandGitHub 2025-10-05 18:09:32 +08:00
• 4b60121498 fix bmnist torch with RANGEIFY=1 (#12442) qazalandGitHub 2025-10-05 12:34:27 +03:00
• b5f31d7505 earlier seen children (#12451) George HotzandGitHub 2025-10-05 15:55:13 +08:00
• dde61c3852 Revert "write tests for algebraic UPat" reprocess_node_experiments geohot 2025-10-05 15:11:07 +08:00
• 136aeaacd3 works geohot 2025-10-05 15:11:05 +08:00
• 18552a3040 canon geohot 2025-10-05 14:58:52 +08:00
• c600446299 experiments with reprocessing node geohot 2025-10-05 14:12:40 +08:00
• 6538935441 write tests for algebraic UPat geohot 2025-10-05 08:20:13 +08:00
• 865d5796f8 add a test for untested Tensor.assign behavior (#12448) qazalandGitHub 2025-10-04 12:44:56 +03:00
• e74be4a140 UOp.factor and add chain sorting (#12413) Sieds LyklesandGitHub 2025-10-04 06:05:38 +02:00
• 394dc24110 post index symbolic (#12446) Sieds LyklesandGitHub 2025-10-03 23:23:03 +02:00
• 9f2b69b870 enable few tests for PTX test_dtype (#12445) chenyuandGitHub 2025-10-03 20:56:30 +08:00
• 0b534f71c2 recursive substitute should be O(n) (#12444) George HotzandGitHub 2025-10-03 18:29:59 +08:00
• 40f0acbc23 make that a single rewrite recurse_sub geohot 2025-10-03 18:18:43 +08:00
• f3ac529438 even faster geohot 2025-10-03 18:14:41 +08:00
• a3991948e9 recursive substitute geohot 2025-10-03 18:03:55 +08:00
• b087663c35 RANGEIFY test_bert uses more ran somehow (#12443) chenyuandGitHub 2025-10-03 16:38:53 +08:00
• 940a8d5ba9 default IGNORE_OOB=1 (#12441) chenyuandGitHub 2025-10-03 16:16:19 +08:00
• d290e77a5b pend substitutes for speed (#12440) George HotzandGitHub 2025-10-03 15:49:19 +08:00
• 23d310bcc1 ptx: handle i8/u8 casts correctly (#12439) nimlgenandGitHub 2025-10-03 10:34:15 +03:00
• 1e8945a28c Training loop for Stable Diffusion mlperf (#12315) hoovedandGitHub 2025-10-03 02:45:38 -04:00
• c7849ac593 fix test lil model (#12437) George HotzandGitHub 2025-10-03 14:28:37 +08:00
• 0f82d92b9d use float for softmax in llm.py (#12438) chenyuandGitHub 2025-10-03 14:27:56 +08:00
• b9f7a7e218 Merge branch 'master' into add_local_buffer George HotzandGitHub 2025-10-03 13:06:14 +08:00
• 4c63f7e786 skip copies of reshaped buffers (#12430) George HotzandGitHub 2025-10-03 13:05:47 +08:00
• 0047bcc535 undo loaded comparison swap (#12436) Sieds LyklesandGitHub 2025-10-03 06:57:29 +02:00
• f203d8b221 update RANGEIFY kernel count and test_masked_select (#12435) chenyuandGitHub 2025-10-03 12:41:34 +08:00
• a6dd5a224b skip webgpu tests (#12433) wozeparrotandGitHub 2025-10-02 21:31:07 -07:00
• bf99de7b1e update a few more tests for RANGEIFY (#12434) chenyuandGitHub 2025-10-03 12:16:58 +08:00
• 9273d7d404 adding a local buffer is very simple now geohot 2025-10-03 11:17:57 +08:00
• a734437da8 skip copies of reshaped buffers geohot 2025-10-03 10:55:58 +08:00
• a8dca47fbc fix support_opts_in_contig geohot 2025-10-03 10:52:18 +08:00
• 769db23df6 that geohot 2025-10-03 10:45:47 +08:00
• 1cd40941c8 delete junk geohot 2025-10-03 10:35:12 +08:00
• 9a607e69e1 Merge branch 'master' into support_opts_in_contig George HotzandGitHub 2025-10-03 10:32:08 +08:00
• 9cd365c12e little changes from double gemm (#12429) George HotzandGitHub 2025-10-03 10:31:51 +08:00
• 16a65b4fd0 fix test_symbolic_gcd_div hang (#12427) Sieds LyklesandGitHub 2025-10-03 04:21:16 +02:00
• 797e74ba1f normal lil_dg_change geohot 2025-10-03 10:12:12 +08:00
• 0f0b0cf17f small changes from double gemm work geohot 2025-10-03 10:07:46 +08:00
• 5b05bf4ab4 Merge branch 'master' into support_opts_in_contig George HotzandGitHub 2025-10-03 10:03:14 +08:00
• 2d24af888b REWRITE_STACK_LIMIT (#12426) chenyuandGitHub 2025-10-03 09:51:04 +08:00
• 1b58ef0d60 Increase stack size limit in unified_rewrite (#12424) hoovedandGitHub 2025-10-02 21:06:47 -04:00
• 17d36d0952 don't tag MSTACK/MSELECT on global buffers (#12423) qazalandGitHub 2025-10-02 13:32:15 +03:00
• e59e0aadc1 opts geohot 2025-10-02 18:25:38 +08:00
• 3688afa513 fix swap geohot 2025-10-02 18:17:03 +08:00
• dae164ffb1 warp geohot 2025-10-02 18:04:45 +08:00
• 3fb3dd4c06 flash attention with two gemms geohot 2025-10-02 17:48:48 +08:00
• e5028d58e9 flash attention sort of works geohot 2025-10-02 17:29:23 +08:00
• 7b3912d8e4 relax atol for some tests (#12422) chenyuandGitHub 2025-10-02 17:04:44 +08:00
• 98163832e4 update RANGEIFY test_cast_padded (#12421) chenyuandGitHub 2025-10-02 16:37:35 +08:00
• 5a602e6c36 double wmma works geohot 2025-10-02 16:06:46 +08:00
• 37beef6de3 add null bert training test in ci (#12420) chenyuandGitHub 2025-10-02 16:05:19 +08:00
• 6640514555 demote works on both matmuls geohot 2025-10-02 15:42:22 +08:00
• f21851b099 ops: n^2 .device property fix (#12419) 3 people 2025-10-02 10:28:12 +03:00
• ec177c80c2 rangeify: fix test_where_fold (llvm) (#12416) b1tgandGitHub 2025-10-02 14:57:49 +08:00
• 13a25b2e67 rangeify: don't shape INDEX on kernelize (#12417) qazalandGitHub 2025-10-02 09:45:37 +03:00
• 5d9035f5a6 Eval for Stable Diffusion mlperf (#12316) hoovedandGitHub 2025-10-02 02:35:38 -04:00
• 0f804c9a83 Stable Diffusion model init for mlperf (#12314) 3 people 2025-10-02 02:28:41 -04:00
• 2fbd7d21f9 tc passes geohot 2025-10-02 14:27:52 +08:00
• 0eee93f0c0 hotfix: disable split ranges for non rangeify geohot 2025-10-02 13:15:24 +08:00
• f32a497f08 bug geohot 2025-10-02 13:10:07 +08:00