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

• 7f7fa26e03 allow hugepage failure in memadvise (#5207) nimlgenandGitHub 2024-06-28 11:41:10 +03:00
• 73395b998b better error msg for TinyJit inside TinyJit (#5202) chenyuandGitHub 2024-06-27 18:09:19 -04:00
• ac748cccdb nv apply relocs (#5165) nimlgenandGitHub 2024-06-27 23:54:16 +03:00
• 540ebdf47c missing init files (#5196) Roelof van DijkandGitHub 2024-06-27 21:30:02 +02:00
• d8dc43ad06 remove JIT_BATCH_SIZE=4 from gpt2 NV benchmark (#5198) chenyuandGitHub 2024-06-27 15:20:34 -04:00
• 345bcc2099 move graph_dedup out of class [run_process_replay] (#5197) George HotzandGitHub 2024-06-27 12:04:00 -07:00
• d094a6828f single pass rewrite (#5159) George HotzandGitHub 2024-06-27 11:36:05 -07:00
• 1ff9bbaa61 ruff: close file handle (#5180) Roelof van DijkandGitHub 2024-06-27 20:29:47 +02:00
• 83da8b3558 use NV instead of CUDA in benchmark (#5192) chenyuandGitHub 2024-06-27 13:52:58 -04:00
• 0c6c7c5f7b CACHELEVEL=0 -> IGNORE_BEAM_CACHE=1 in benchmark (#5191) chenyuandGitHub 2024-06-27 13:15:18 -04:00
• c12de4f47d benchmark use JITBEAM for llama and gpt2 (#5189) chenyuandGitHub 2024-06-27 12:56:02 -04:00
• ad91962dcf CACHECOLLECTING -> CAPTURING and don't capture clear_l2 (#5190) chenyuandGitHub 2024-06-27 12:32:28 -04:00
• 01e8838b65 ruff: suppressible-exception (#5182) 3 people 2024-06-27 17:23:44 +02:00
• 9704c7d4d4 ruff rule if-exp-instead-of-or-operator (FURB110) (#5178) 3 people 2024-06-27 17:22:19 +02:00
• 5b8fda3c65 fix: JIT=0 means no JIT (#5188) chenyuandGitHub 2024-06-27 10:31:37 -04:00
• 3af17849bf safely parse quoted titles [run_process_replay] (#5183) qazalandGitHub 2024-06-27 16:39:48 +03:00
• 975b811ad9 names shadowing builtins (#5179) 3 people 2024-06-27 14:15:01 +02:00
• 26e254c42b ruff: else-raise and else-return (#5175) 3 people 2024-06-27 13:54:59 +02:00
• f88f71d73a ruff: unnecessary-comprehension (#5174) Roelof van DijkandGitHub 2024-06-27 13:45:29 +02:00
• f1c7944c44 Fix batchnorm shapes for resnet.load_pretrained (#5167) reddyn12andGitHub 2024-06-26 18:44:10 -04:00
• 396ce6cfc9 clean up graph dedup function [run_process_replay] (#5169) George HotzandGitHub 2024-06-26 15:07:34 -07:00
• 3a04e518ec print_tree UPat +fix (#5132) kormannandGitHub 2024-06-27 00:02:19 +02:00
• 0ba093dea0 hotfix: only validate stable diffusion when using threefry (#5166) chenyuandGitHub 2024-06-26 16:50:38 -04:00
• e4a5870b36 validate stable_diffusion output (#5163) chenyuandGitHub 2024-06-26 16:42:21 -04:00
• 21b225ac45 llama3 download works (#5160) nimlgenandGitHub 2024-06-26 22:45:13 +03:00
• c91b3c4079 shard llama3 on 0 sometimes (#5157) wozeparrotandGitHub 2024-06-26 18:50:57 +00:00
• 294bd1a9ff refactor: name check [run_process_replay] (#5158) Roelof van DijkandGitHub 2024-06-26 20:39:41 +02:00
• 2c80583e14 perf: cache const UOp creation [run_process_replay] (#5156) Roelof van DijkandGitHub 2024-06-26 20:13:14 +02:00
• eda2824cd8 freeze uop [run_process_replay] (#5155) George HotzandGitHub 2024-06-26 10:18:15 -07:00
• e267f3161d Add MLLogger (#5125) Elias WahlandGitHub 2024-06-26 18:23:56 +02:00
• 16405b973a fix hcq sync (#5062) nimlgenandGitHub 2024-06-26 17:50:37 +03:00
• 3604642847 Llama shard axis 0 sometimes (#5123) 4 people 2024-06-26 07:35:25 -07:00
• fd27f19e92 graph tests (#5153) nimlgenandGitHub 2024-06-26 16:31:20 +03:00
• 7b709c3ccd switch tensorcoreoptions to tuple [run_process_replay] (#5143) 3 people 2024-06-26 04:12:53 -07:00
• 6ca7b13ed1 limit pickled objects [run_process_replay] (#5154) qazalandGitHub 2024-06-26 13:51:32 +03:00
• ee4f080a14 rewrite div const [run_process_replay] [no_assert] (#5151) George HotzandGitHub 2024-06-25 20:23:14 -07:00
• 666a9c1448 don't view origin buffer when sharding (#5122) David HouandGitHub 2024-06-25 20:19:09 -07:00
• 89e106686a simpler unmatch [run_process_replay] (#5149) George HotzandGitHub 2024-06-25 19:57:40 -07:00
• c98ca23cb9 test pickle variable (#5150) George HotzandGitHub 2024-06-25 19:49:21 -07:00
• 8fcc41582f make buffer view optional with a flag (#5120) David HouandGitHub 2024-06-25 19:13:20 -07:00
• 63ba2d05d1 uops dfs cleanup (#5147) George HotzandGitHub 2024-06-25 18:51:42 -07:00
• 6841ea3baf don't allow duplicate variables (#5148) George HotzandGitHub 2024-06-25 18:47:29 -07:00
• cc7fafcd8b sink folding rule [run_process_replay] (#5145) George HotzandGitHub 2024-06-25 18:34:44 -07:00
• fa78755f19 Add new patterns to unfold division (#5139) Jhenner TigrerosandGitHub 2024-06-25 20:07:47 -05:00
• c4fdb9c725 second iteration on verify_lazyop (#5140) qazalandGitHub 2024-06-25 09:44:32 +03:00
• dade7677cf validate llama3 output only with model "LLaMA-3/8B-SF-DPO" (#5138) chenyuandGitHub 2024-06-24 20:58:25 -04:00
• 981afb114f safely fold NEG in lazy.py (#5135) qazalandGitHub 2024-06-25 02:40:37 +03:00
• 7948b05738 fix uneven shard with shrink and pad args on sharded axis (#5131) chenyuandGitHub 2024-06-24 16:55:50 -04:00
• 18e70deec3 verify_lazyop (#5124) qazalandGitHub 2024-06-24 23:45:35 +03:00
• fe707bc968 hotfix: don't use is for comparing dtype (#5128) qazalandGitHub 2024-06-24 21:12:34 +03:00
• dfa562dbc1 DEFINE_ACC takes UOps.CONST in vin instead of arg (#4975) 3 people 2024-06-24 11:25:33 -05:00
• d84beaa6dd tiny profiler cleanups (#5126) nimlgenandGitHub 2024-06-24 17:02:31 +03:00
• 4a7d403777 cleanup test_multitensor (#5118) chenyuandGitHub 2024-06-23 20:54:22 -04:00
• c0ba5e0dfb multi copy_to_device return the copy on same device if possible (#5117) chenyuandGitHub 2024-06-23 20:25:56 -04:00
• b563cd52ed linearizer: change globals to merge into left axis/gridDims.x first (#5033) Francis LamandGitHub 2024-06-23 15:53:15 -07:00
• 69f116a7e1 nv/amd profiler (#4718) nimlgenandGitHub 2024-06-23 17:10:12 +03:00
• 64a3b7931e simplify render_ops ctx [run_process_replay] (#5116) qazalandGitHub 2024-06-23 16:56:32 +03:00
• 28bf8d86d8 test_linearizer with multi output ASTs (#5115) qazalandGitHub 2024-06-23 15:41:24 +03:00
• ee0c6dfc15 build Tensor._tri with movements only (#5110) chenyuandGitHub 2024-06-23 00:07:36 -04:00
• 20fabd8a5b update Tensor.triu and Tensor.tril (#5109) chenyuandGitHub 2024-06-22 21:59:50 -04:00
• 8f6ae84e4a minor cleanup of conv_transpose2d (#5108) chenyuandGitHub 2024-06-22 21:31:47 -04:00
• 33211f356b fix desc in tqdm (#5107) chenyuandGitHub 2024-06-22 19:00:38 -04:00
• 055e616302 cleanup mnist data load in beautiful_mnist (#5106) chenyuandGitHub 2024-06-22 18:31:51 -04:00
• 5516b790ad hotfix append colon space to tqdm set_description (#5105) chenyuandGitHub 2024-06-22 18:09:14 -04:00
• e356807696 tinytqdm.set_description and tinytrange (#5101) chenyuandGitHub 2024-06-22 14:45:06 -04:00
• 8080298739 s/tinytqdm/tqdm (#5103) chenyuandGitHub 2024-06-22 14:18:26 -04:00
• 9f875123b6 small changes from lowerer. [run_process_replay] [no_assert] (#5102) George HotzandGitHub 2024-06-22 11:09:35 -07:00
• e468601226 update llama attention casting (#5096) chenyuandGitHub 2024-06-22 10:57:17 -04:00
• ca021229e4 fix attention to always return in the same dtype as input (#5100) chenyuandGitHub 2024-06-22 10:34:57 -04:00
• 2dcef5a0d7 hcq spec (#5081) nimlgenandGitHub 2024-06-22 15:32:12 +03:00
• 8bd6cb9511 update llama model RMSNorm casting (#5095) chenyuandGitHub 2024-06-21 23:02:04 -04:00
• 0c857ae2d6 some onnx_ops cleanups (#5094) chenyuandGitHub 2024-06-21 22:01:32 -04:00
• f4a041af16 Simplify graph_dedup [run_process_replay] (#5084) kormannandGitHub 2024-06-21 21:12:30 +02:00
• 00593d6095 clean the long lines in avg_pool2d and max_pool2d (#5091) chenyuandGitHub 2024-06-21 14:46:56 -04:00
• a971dc6218 argmax(axis=None) is argmax.flatten().argmax(0) (#5090) chenyuandGitHub 2024-06-21 14:17:10 -04:00
• 166a2b19b5 fix reduce axis of 0d tensors (#5089) chenyuandGitHub 2024-06-21 13:51:40 -04:00
• 3ff048b68c type annotate reduce axis in tensor.py (#5088) chenyuandGitHub 2024-06-21 13:06:10 -04:00
• 36b4a492a1 explicitly check getitem indices can have at most one ellipsis (#5087) chenyuandGitHub 2024-06-21 12:33:18 -04:00
• f1e758bacb graph fuzzer (#5082) nimlgenandGitHub 2024-06-21 18:47:23 +03:00
• 5717a54b28 don't use Tensor.empty in kernel opts tests (#5086) qazalandGitHub 2024-06-21 18:41:03 +03:00
• 8aa786232d docs for running process replay locally (#5083) qazalandGitHub 2024-06-21 16:55:08 +03:00
• fb1bf48cfe io_uring for copies from disk (#5035) nimlgenandGitHub 2024-06-21 11:36:51 +03:00
• b69afc67d8 tinybox docs typo geohot 2024-06-20 17:58:40 -07:00
• 6bc5e5f41c start tinybox docs geohot 2024-06-20 17:04:45 -07:00
• f6d6760f71 don't cast tuple to list before creating Tensor (#5071) chenyuandGitHub 2024-06-20 13:32:56 -04:00
• 97f1347dd9 fix check_process_replay for special characters (#5072) qazalandGitHub 2024-06-20 20:23:29 +03:00
• 6f6b3b10c9 import from uops, not linearizer (#5064) George HotzandGitHub 2024-06-20 08:08:44 -07:00
• 50700171ef minor cleanup to reshape arg handling (#5070) chenyuandGitHub 2024-06-20 10:27:27 -04:00
• f4355d0f1b check Tensor.permute input arg is a valid permutation (#5069) chenyuandGitHub 2024-06-20 10:01:28 -04:00
• 24c89a2a33 move assert_equiv_uops to helpers + use == for dtypes (#5067) qazalandGitHub 2024-06-20 16:39:34 +03:00
• e8f39fcaaa check arg to Tensor.flip can appear only once (#5068) chenyuandGitHub 2024-06-20 09:33:42 -04:00
• 55e02cdd84 generic gate folding (#5061) qazalandGitHub 2024-06-20 16:10:08 +03:00
• bdca2da2be typannos (#5059) kormannandGitHub 2024-06-20 15:02:31 +02:00
• 5f7edc7a46 minor cleanup getting output shape of _pool (#5065) chenyuandGitHub 2024-06-20 09:00:48 -04:00
• a6a5dba637 Revert "UPat for has_valid in load/store (#5052)" (#5056) qazalandGitHub 2024-06-19 20:53:36 +03:00
• ee01e464e3 use process replay as a diff creator (#4903) qazalandGitHub 2024-06-19 18:17:31 +03:00
• 99fc275c27 UPat line savings [run_process_replay] (#5053) qazalandGitHub 2024-06-19 12:43:20 +03:00
• 71194df1da UPat for has_valid in load/store [run_process_replay] (#5052) qazalandGitHub 2024-06-19 12:20:22 +03:00
• 996788358d minor change to gelu (#5048) chenyuandGitHub 2024-06-18 22:26:56 -04:00
• 4c7e316ded update pylint for ops_python (#5046) chenyuandGitHub 2024-06-18 20:15:34 -04:00