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

autogen-comgr-2

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

ci-deps

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

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

explicit-dev

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

gitea-test-ci

gl_dims

gldims

glm_flash

grad_outerworld

grok_x86_novec

group_div_rules

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-chat-template

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

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

onnx-cpu-count

outer_range

outerworld_work

pad_invalid

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

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-lp

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

view_after_const

view_in_codegen

viz_cbroadcast

warp_fun

wgpu-f16-packed

working_postopt

x86_moves

x86_numel

#1

#2

#3

#4

#5

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

• e53673a0b2 amd: sdma queue overrun fix (#11012) nimlgenandGitHub 2025-06-28 01:42:03 +03:00
• f2548afeb5 bert grad clipping start with const 0 (#11008) chenyuandGitHub 2025-06-27 18:02:23 -04:00
• a6485d00c8 very tiny generate_dataset (#11013) chenyuandGitHub 2025-06-27 17:10:45 -04:00
• 382fa6a325 viz: support axis colors in UOp nodes (#11009) qazalandGitHub 2025-06-27 23:02:55 +03:00
• 44257f25e4 bump line count to 14600 (#11010) qazalandGitHub 2025-06-27 22:48:14 +03:00
• be53ef4f0a rename DEFINE_ACC -> DEFINE_REG (#11006) George HotzandGitHub 2025-06-27 11:09:25 -07:00
• 05c35d0db8 reorder ops and add comments (#11005) George HotzandGitHub 2025-06-27 10:52:14 -07:00
• 5a1911b7c4 apply the global dims late (#11002) George HotzandGitHub 2025-06-27 09:54:34 -07:00
• 4ef10c57f9 remove unused test helper (#10999) qazalandGitHub 2025-06-27 13:48:48 +03:00
• a39343e39f viz: move timeline layout to python (#10998) qazalandGitHub 2025-06-27 13:06:00 +03:00
• b4eb876d5a kernel.py no longer permutes reduce axis [pr] (#10968) George HotzandGitHub 2025-06-26 17:44:58 -07:00
• 6ab5a5cb6c llama3 mlperf train (#10983) chenyuandGitHub 2025-06-26 20:24:27 -04:00
• 856759c79c add halide example (#10980) George HotzandGitHub 2025-06-26 16:14:57 -07:00
• 1127302c46 move perfetto to extra (#10994) qazalandGitHub 2025-06-27 01:53:54 +03:00
• 712980e167 fix extract_dataset + add tests to CI (#10995) qazalandGitHub 2025-06-27 01:51:36 +03:00
• 4572e65f0f remove duplicated move_early logic in UOp.r [pr] (#10993) chenyuandGitHub 2025-06-26 18:33:54 -04:00
• 579194f523 remove some linearize calls from tests 2 [pr] (#10992) Ignacio SicaandGitHub 2025-06-26 18:22:27 -03:00
• 50936b4a18 ONNX real float16 (#10694) 3 people 2025-06-27 02:05:12 +08:00
• 73484b0803 viz: generic shape tooltip/click handlers + renames (#10990) qazalandGitHub 2025-06-26 19:14:04 +03:00
• 7f79c1388f viz: update y offset calculation (#10987) qazalandGitHub 2025-06-26 12:05:20 +03:00
• 49bba2f0a0 improve test_nll_loss (#10986) chenyuandGitHub 2025-06-26 02:46:55 -04:00
• 0612acfc70 improve Tensor.cross_entropy (#10985) chenyuandGitHub 2025-06-26 01:39:48 -04:00
• 8751d47985 CosineAnnealingLRWithWarmup (#10981) chenyuandGitHub 2025-06-25 17:45:21 -04:00
• 21f1c4cc09 remove some linearize calls from tests [pr] (#10978) Ignacio SicaandGitHub 2025-06-25 16:37:17 -03:00
• efad567ebd ruff check whole examples/mlperf/ (#10979) chenyuandGitHub 2025-06-25 12:57:48 -04:00
• 15e60caf09 add Ops.EQ (#10976) Sieds LyklesandGitHub 2025-06-25 17:25:10 +02:00
• 98d2cde293 revert tc_group feature (#10971) Ignacio SicaandGitHub 2025-06-25 00:58:13 -03:00
• 306dbc76f6 early view simplify (#10974) George HotzandGitHub 2025-06-24 20:52:45 -07:00
• 77fff73295 fix viz vscode link on windows (#10972) 3 people 2025-06-25 11:47:59 +08:00
• 9d995c2a4d shape const if it has a device [pr] (#10969) George HotzandGitHub 2025-06-24 16:22:54 -07:00
• cf60ccac6a support new const lowering (#10967) George HotzandGitHub 2025-06-24 15:21:41 -07:00
• 8a65720528 hotfix: disable test_tensor_core_opts_group test on real metal geohot 2025-06-24 15:21:33 -07:00
• 1c45b9f7fb start nvpci (#10521) nimlgenandGitHub 2025-06-25 00:37:34 +03:00
• c8d0f68763 Weaker renderer validation in remote (#10964) uuuvnandGitHub 2025-06-24 21:15:09 +00:00
• c2f5f0f198 more robust reduce_gradient (#10965) George HotzandGitHub 2025-06-24 14:09:33 -07:00
• 8743ca40e2 force reduce to be in axis order (#10837) George HotzandGitHub 2025-06-24 13:00:16 -07:00
• ffb032e31d test_diagonal touchup (#10962) chenyuandGitHub 2025-06-24 15:51:19 -04:00
• 7f9958b632 Fix torch.linalg.diagonal crash due to invalid shrink in to_movement_ops (#10945) 3 people 2025-06-25 01:06:06 +05:30
• 26ddf8d714 amd: rename dev_iface -> iface to match nv (#10959) nimlgenandGitHub 2025-06-24 20:22:19 +03:00
• bfa87f3490 clean up binary_crossentropy_logits (#10958) chenyuandGitHub 2025-06-24 12:23:40 -04:00
• 2ccddfc0ca viz: match canvas fontsize (#10957) qazalandGitHub 2025-06-24 19:07:06 +03:00
• de4b9bf53b add opts_to_apply option to AST KernelInfo (#10950) qazalandGitHub 2025-06-24 18:55:39 +03:00
• 18e264a449 Tensor.logsigmoid (#10955) chenyuandGitHub 2025-06-24 11:16:14 -04:00
• f15247d2d2 remove outdated index masking in lowerer [pr] (#10953) Ignacio SicaandGitHub 2025-06-24 11:53:30 -03:00
• cc32394b32 support copyin/copyout/is_allocated for subbuffers (#10869) 4 people 2025-06-24 22:49:04 +08:00
• 35504c938e torch.clip(x,y) -> x.clip(y) in test_ops (#10954) chenyuandGitHub 2025-06-24 10:22:19 -04:00
• 86d458533f Add pos_weight for binary_crossentropy_logits (#10855) Fang-Pen LinandGitHub 2025-06-24 06:42:37 -07:00
• 61dad3740f fix min_max and add test (#10952) Sieds LyklesandGitHub 2025-06-24 15:33:26 +02:00
• ab8c5d04ab viz: convert to function_name in server [pr] (#10951) qazalandGitHub 2025-06-24 13:59:37 +03:00
• c0d9cf09e0 system: flock (#10949) nimlgenandGitHub 2025-06-24 11:33:49 +03:00
• 5202970feb system: move memory_barrier to System (#10948) nimlgenandGitHub 2025-06-24 11:09:43 +03:00
• f41c28a048 update test_tensor_uop_representation comments [pr] (#10946) qazalandGitHub 2025-06-24 10:47:09 +03:00
• 7a5e4e0bf1 fix unittests process replay [pr] (#10947) qazalandGitHub 2025-06-24 10:30:23 +03:00
• 7d560dbd75 hotfix: corealize in the tiny mnist test geohot 2025-06-23 17:40:33 -07:00
• 230ad3a460 [bounty] Don't use numpy inside hlb_cifar10 training loop (#10777) 3 people 2025-06-23 21:24:56 -03:00
• 383010555f delete linearize and to_program from kernel.py (#10943) George HotzandGitHub 2025-06-23 17:04:05 -07:00
• 0f89660ce4 Revert "change clang -march flag to -mcpu on arm (#10841)" (#10942) George HotzandGitHub 2025-06-23 16:48:28 -07:00
• 956a8391a5 minor cleanup on test_tensor_core_opts tests (#10924) 3 people 2025-06-23 20:30:21 -03:00
• 897e42fd1b change clang -march flag to -mcpu on arm (#10841) ttomsaandGitHub 2025-06-24 00:28:48 +01:00
• 772cd02ad2 Perform index validation on load/store, not on the index (#10849) Sieds LyklesandGitHub 2025-06-24 01:25:05 +02:00
• ae4d2d71b4 bump line count to 14500 geohot 2025-06-23 15:29:03 -07:00
• 79d7cdd9ba Fix device (#10929) Harsh NatuskarandGitHub 2025-06-24 04:00:19 +05:30
• e15754db28 remove (some) kernelize from llama and test schedule speed (#10939) George HotzandGitHub 2025-06-23 15:07:31 -07:00
• 3699d1d3ba hotfix llama3 temperature is float (#10938) chenyuandGitHub 2025-06-23 15:20:56 -04:00
• 4e2c9e36c7 Remote multihost (p2p transfer) (#10601) uuuvnandGitHub 2025-06-23 18:47:29 +00:00
• 42b1c9625b skip test TestKiTS19Dataset::test_training_set (#10936) chenyuandGitHub 2025-06-23 14:27:24 -04:00
• 9e9fd44987 refactor test/external/external_llama_eval.py (#10567) 3 people 2025-06-23 20:43:20 +03:00
• 785b4ea8ac optim flatten().shape[0] is numel (#10935) chenyuandGitHub 2025-06-23 13:11:19 -04:00
• ac39f27ae6 viz: non blocking UOp tracing (#10913) qazalandGitHub 2025-06-23 19:59:28 +03:00
• b8d09a1dae tc with group/grouptop (#10903) Ignacio SicaandGitHub 2025-06-23 13:58:41 -03:00
• 9944c2c02d viz: show time taken on hover (#10934) qazalandGitHub 2025-06-23 19:00:40 +03:00
• 1e99a7f1c9 hotfix: don't viz the indexing rewrites geohot 2025-06-23 08:20:26 -07:00
• f9b59924f1 OPTIM_DTYPE to specify dtype for optim params (#10925) chenyuandGitHub 2025-06-23 10:32:03 -04:00
• 7820aeca8e update codegen process replay to use get_program [pr] (#10921) qazalandGitHub 2025-06-23 17:31:41 +03:00
• eceb7a00d2 nv: rename iface mem functions (#10931) nimlgenandGitHub 2025-06-23 16:34:51 +03:00
• 4e864bd304 fix: getenv("NOLOCALS")/NOLOCALS context var (#10927) qazalandGitHub 2025-06-23 11:23:59 +03:00
• 22f9696522 Fix/hcqfuzz harnesss bug (#10923) alpharushandGitHub 2025-06-23 03:22:30 -05:00
• f037f85532 s/getenv("TC")/USE_TC context var (#10922) qazalandGitHub 2025-06-23 00:39:45 +03:00
• 9201224e0b viz: remove Kernel check [pr] (#10920) qazalandGitHub 2025-06-22 20:47:54 +03:00
• 3ccdb2356b system: factor out PCIIfaceBase (#10917) nimlgenandGitHub 2025-06-22 20:03:14 +03:00
• b09c47366f opt transforms the ast into an optimized ast (#10900) George HotzandGitHub 2025-06-22 09:41:26 -07:00
• ffddf165f8 viz: color by kernel names in profiler (#10919) qazalandGitHub 2025-06-22 18:07:52 +03:00
• 36536ef6f0 nv: minor changes from nvpci (#10918) nimlgenandGitHub 2025-06-22 18:04:39 +03:00
• 4ab7d792cc ONNX improve dtype fallback (#10800) geohotstanandGitHub 2025-06-22 07:29:45 +08:00
• 0480139def log_perplexity metrics (#10912) chenyuandGitHub 2025-06-21 10:44:47 -04:00
• 0e7bd9fd03 factor out generic MemoryManager (#10910) nimlgenandGitHub 2025-06-21 16:18:33 +03:00
• c7ec913210 viz: cleanup unit tests (#10909) qazalandGitHub 2025-06-21 12:35:09 +03:00
• 1373071f19 simplify logcumsumexp (#10908) chenyuandGitHub 2025-06-20 22:56:42 -04:00
• fa52bdb50f applied_opts is in the optimized ast] (#10906) George HotzandGitHub 2025-06-20 18:56:23 -07:00
• 2d9c61e39e test more dims in test_logsumexp and test_logcumsumexp (#10907) chenyuandGitHub 2025-06-20 21:42:18 -04:00
• 3771cc0f77 fix test logcumsumexp broken devectorize=0 (#10880) Nino RisteskiandGitHub 2025-06-21 02:54:50 +02:00
• 7636d2cdc5 flip order of get_program args (#10905) George HotzandGitHub 2025-06-20 17:23:23 -07:00
• 1ce63f8d04 move functions to view and update docs [pr] (#10904) George HotzandGitHub 2025-06-20 16:47:58 -07:00
• 04a8eca8e3 add TestKernelOpts.test_tensor_core_opts run in benchmarks update_benchmark_search ignaciosica 2025-06-20 20:10:44 -03:00
• b41e0563a3 move stuff to kernelize folder (#10902) George HotzandGitHub 2025-06-20 16:10:20 -07:00
• d399a4587d move mem estimate to ProgramSpec [pr] (#10901) George HotzandGitHub 2025-06-20 15:54:28 -07:00
• 210b847b44 init tc with group ignaciosica 2025-06-20 15:33:32 -03:00
• 92678e59ee move kernel to opt (#10899) George HotzandGitHub 2025-06-20 15:22:28 -07:00
• bb0299b9e5 system: shared pci logic (#10894) nimlgenandGitHub 2025-06-21 00:09:49 +03:00
• c83fdc50d1 nv: driver iface (#10895) nimlgenandGitHub 2025-06-20 22:36:08 +03:00