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

• dfbee4f0f5 feat: add blobfile to testing (#5254) wozeparrotandGitHub 2024-07-02 02:33:58 +00:00
• 8c9c1cf62f Pulled CLIP and UNet into Seperate Files (#5253) Tobias FischerandGitHub 2024-07-01 22:33:01 -04:00
• 5808c37302 hotfix disable flaky llama3 beam benchmark on green (#5249) chenyuandGitHub 2024-07-01 15:00:47 -04:00
• b9122ecdaf revert stable diffusion validation with threefry (#5248) chenyuandGitHub 2024-07-01 14:43:47 -04:00
• 57e89645cd hcq spec test (#5226) nimlgenandGitHub 2024-07-01 17:36:37 +03:00
• d7839fdc5f Add x!=0 -> (bool)x pattern [run_process_replay] [no_assert] (#5237) Carson PowersandGitHub 2024-06-30 17:48:45 -07:00
• 14980f79dd hotfix: unbreak llama geohot 2024-06-30 15:27:54 -07:00
• 146eb3a811 hotfix: add repeat_interleave docs geohot 2024-06-30 15:25:18 -07:00
• 3df47bc21e OpenELM + repeat_interleave (#5234) George HotzandGitHub 2024-06-30 15:18:39 -07:00
• 7b7b751513 simple hip backend for debugging (#5201) nimlgenandGitHub 2024-06-30 23:00:11 +03:00
• 88763eb9ff fix stable_diffusion with fp16 (#5239) chenyuandGitHub 2024-06-30 12:59:31 -04:00
• 649641a2f2 fix tqdm with generator without __len__ (#5238) chenyuandGitHub 2024-06-30 12:20:59 -04:00
• fd53b6d901 tqdm supports fractional blocks (#5233) chenyuandGitHub 2024-06-29 22:30:18 -04:00
• ae10ae4722 simplify tqdm scale math (#5231) chenyuandGitHub 2024-06-29 21:17:40 -04:00
• ad1ca7da64 [Feature] Added BinaryOps.AND/BinaryOps.OR (#5223) hiketteiandGitHub 2024-06-30 09:20:25 +09:00
• 50b05dd3f4 tqdm minor cleanup (#5229) chenyuandGitHub 2024-06-29 18:58:24 -04:00
• b2ea610df8 fix tqdm unit_scale and support hours in time (#5227) chenyuandGitHub 2024-06-29 14:48:51 -04:00
• f374fb77af assert bool dtype for valid [run_process_replay] (#5214) qazalandGitHub 2024-06-29 21:20:32 +03:00
• 3f4eeb8b54 late UOps.IF generation [run_process_replay] [no_assert] (#5027) qazalandGitHub 2024-06-29 19:22:14 +03:00
• 42d1f92fc1 simpler tqdm (#5221) chenyuandGitHub 2024-06-29 07:41:46 -04:00
• dd7eef7d71 libc defs to autogen (#5217) nimlgenandGitHub 2024-06-29 14:37:33 +03:00
• 6b08cb5e38 ptx runs on nv in benchmarks (#5224) nimlgenandGitHub 2024-06-29 11:06:44 +03:00
• b4c49ae3fa remove cudacpu in favour of mockgpu (#5225) nimlgenandGitHub 2024-06-29 11:05:16 +03:00
• ee02dcb98e nv supports PTX=1 (#5222) nimlgenandGitHub 2024-06-29 10:46:29 +03:00
• 7bcb74ab23 feat: tag 0.9.1 (#5220) v0.9.1 wozeparrotandGitHub 2024-06-29 03:16:14 +00:00
• 7f46bfa587 hotfix: docs touchup geohot 2024-06-28 14:36:20 -07:00
• c941a58581 amd refactor queue creation (#5216) nimlgenandGitHub 2024-06-28 23:24:49 +03:00
• 7ba4938510 simplify View.permute arg check [run_process_replay] (#5218) chenyuandGitHub 2024-06-28 16:18:46 -04:00
• 80ac21200b hotfix: linearizer test fixup geohot 2024-06-28 10:52:25 -07:00
• c9714dfcf4 rename graph to children [run_process_replay] (#5215) George HotzandGitHub 2024-06-28 09:53:52 -07:00
• 6c456b6d66 remove uopgraph dedup + slight speedup (#5199) kormannandGitHub 2024-06-28 18:26:32 +02:00
• 9b08a9397c amd inline bf16 funcs (#5212) nimlgenandGitHub 2024-06-28 18:45:00 +03:00
• 7090eac8cb validate sdxl output and put it in benchmark (#5211) chenyuandGitHub 2024-06-28 11:40:52 -04:00
• 63fa4e2a0e fix seed = 0 in sdxl (#5209) chenyuandGitHub 2024-06-28 08:48:59 -04:00
• 4688f97d48 Add SDXL Inference to Examples (#5206) Tobias FischerandGitHub 2024-06-28 07:42:28 -04:00
• 3e56c8422c remu err handling (#5208) qazalandGitHub 2024-06-28 13:15:18 +03:00
• 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