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

• 90332eb529 Getitem pin None dimension (#4960) geohotstanandGitHub 2024-06-14 22:48:59 +08:00
• 2eeddf1a46 IF ends with STORE, RANGE ends with PHI [run_process_replay] (#4953) qazalandGitHub 2024-06-14 21:00:32 +08:00
• d5a92b9b83 sort the axis in reduce op [run_process_replay] (#4956) George HotzandGitHub 2024-06-14 05:16:05 -07:00
• 14189bca68 graph_dedup function [run_process_replay] (#4955) George HotzandGitHub 2024-06-14 04:24:37 -07:00
• 63a8add2c2 move uops add logic to linearize (#4952) George HotzandGitHub 2024-06-14 03:52:37 -07:00
• 7e32b8c930 refactor generic UOps.END* insertion (#4951) qazalandGitHub 2024-06-14 18:42:41 +08:00
• 9823752397 make uops.add private (#4950) George HotzandGitHub 2024-06-14 03:23:25 -07:00
• dc9e9e4363 Convert BinaryOps.DIV to UnaryOps.RECIP and BinaryOps.IDIV (#4887) 3 people 2024-06-14 04:43:46 -05:00
• f87ba6016a tqdm total=0 fix (#4939) SnakeOnexandGitHub 2024-06-14 11:31:59 +02:00
• 225f792330 amd hdp flush regs are on seg2 (#4925) nimlgenandGitHub 2024-06-14 01:42:23 +03:00
• 4bfd1904f6 nv do not modify prg's qmd (#4948) nimlgenandGitHub 2024-06-14 01:15:40 +03:00
• 845c10bc28 add Node to _broadcasted type annotation (#4946) chenyuandGitHub 2024-06-13 14:10:56 -04:00
• 287d3c3b84 support list, tuple input in dtypes.from_py (#4945) chenyuandGitHub 2024-06-13 13:38:06 -04:00
• 7aecea4f56 support creating Tensor from python tuple (#4944) chenyuandGitHub 2024-06-13 12:18:37 -04:00
• 74586bc339 fix getitem with leading None (#4943) chenyuandGitHub 2024-06-13 11:23:40 -04:00
• e63701fbd4 RDNA3 assembly support (#3637) George HotzandGitHub 2024-06-13 09:09:24 +02:00
• fd071ba27e amd mockgpu correct timer resolution (#4942) nimlgenandGitHub 2024-06-13 10:07:34 +03:00
• fae08c4d48 fix Tensor.triu / Tensor.triu with boolean input (#4941) chenyuandGitHub 2024-06-12 20:16:13 -04:00
• cc90b3ef9f simpler Tensor.gather (#4940) chenyuandGitHub 2024-06-12 19:42:40 -04:00
• fa00ef66fd Update README.md George HotzandGitHub 2024-06-13 00:29:19 +02:00
• eb0f5b5660 failed test case for getitem with leading Nones (#4936) chenyuandGitHub 2024-06-12 16:19:42 -04:00
• d2e3c391e8 Residual in MLM loss + Change default steps (#4935) Elias WahlandGitHub 2024-06-12 22:09:18 +02:00
• a21ea165bc skip linearizer test_failure_22 on llvm (#4937) chenyuandGitHub 2024-06-12 16:03:38 -04:00
• 27903c5ed5 minor minor Tensor.__getitem__ cleanup (#4934) chenyuandGitHub 2024-06-12 15:08:18 -04:00
• 5e6336edda minor Tensor.gather cleanup (#4933) chenyuandGitHub 2024-06-12 13:57:28 -04:00
• 720c700a8a Multireduce-Kernels: Linearizer Changes and Tests (#4259) 5 people 2024-06-12 10:29:43 -07:00
• 6e86472cd6 fix typing for test to run in py38 (#4930) Nicklas BomanandGitHub 2024-06-12 19:22:30 +02:00
• 1326f29e24 fix Tensor.gather shape checking criteria (#4932) chenyuandGitHub 2024-06-12 13:10:14 -04:00
• 898430c004 more typing in linearizer uoping utils (#4929) qazalandGitHub 2024-06-12 23:00:02 +08:00
• 828c98d5c4 add slides from code europe to docs geohot 2024-06-12 14:35:08 +02:00
• 9a3c1e4a17 fix mul div failure (#4928) George HotzandGitHub 2024-06-12 13:58:46 +02:00
• 11a03cbbf5 don't use uops.add while constructing (#4913) George HotzandGitHub 2024-06-12 13:31:34 +02:00
• d894acbb50 remove hardcoded -1s referencing late reduce (#4926) qazalandGitHub 2024-06-12 16:50:15 +08:00
• b833a112ba allocate shared memory per block (#4924) qazalandGitHub 2024-06-12 15:43:10 +08:00
• ca4ccddcd6 docsfix: nn.Tensor -> Tensor geohot 2024-06-12 09:16:24 +02:00
• 3d13c23bfa llama3 --download_model (#4922) wozeparrotandGitHub 2024-06-12 05:59:59 +00:00
• f902af4f0b increase metal ci test timeout to 20 minutes (#4920) chenyuandGitHub 2024-06-11 18:45:51 -04:00
• fdbb4305cb skip unsupported dtype in fuzz_linearizer (#4917) chenyuandGitHub 2024-06-11 18:18:21 -04:00
• 7f3d9e6d94 revert hsa autogen removal (#4914) qazalandGitHub 2024-06-12 00:55:45 +08:00
• 58cf6eaba9 add missing dir level for amd mockgpu (#4911) nimlgenandGitHub 2024-06-11 19:35:04 +03:00
• b886d250fb improve test_dropout_on_shard (#4912) chenyuandGitHub 2024-06-11 11:36:02 -04:00
• 7f03420d05 only install comgr in AMD CI (#4909) qazalandGitHub 2024-06-11 18:19:33 +08:00
• 35e53c0809 add sharded arange test (#4908) George HotzandGitHub 2024-06-11 10:58:33 +02:00
• 798ea61377 widen test_ops [low, high] and more strict atol (#4906) chenyuandGitHub 2024-06-10 20:47:09 -04:00
• 97b05f567e revert the .detach() in layernorm (#4904) chenyuandGitHub 2024-06-10 18:02:05 -04:00
• 8b5bcf309a process replay in all of CI (#4884) qazalandGitHub 2024-06-11 02:49:29 +08:00
• 9715a7193a replace set with dedup (#4901) George HotzandGitHub 2024-06-10 18:20:38 +02:00
• c8cd637236 test case for Tensor.var reducing over size = 1 axis (#4902) chenyuandGitHub 2024-06-10 12:11:39 -04:00
• c0fb7eee09 cleanup lazy const fold for binary (#4900) chenyuandGitHub 2024-06-10 10:46:58 -04:00
• 5bf1f7d4d3 nv better error messages for ioctls (#4899) nimlgenandGitHub 2024-06-10 16:01:50 +03:00
• b9f26eedc9 hotfix: import datasets in nn init geohot 2024-06-10 11:33:50 +02:00
• b56ae5606c cosmetic changes to uop _match (#4897) chenyuandGitHub 2024-06-09 18:29:42 -04:00
• b1db2d0094 tqdm replacement (#4846) SnakeOnexandGitHub 2024-06-09 23:46:03 +02:00
• 05d7ab774f set tensor core opt options in Renderer (#4896) qazalandGitHub 2024-06-10 02:12:41 +08:00
• f42183ba28 hotfix: relax cifar to 93.2 geohot 2024-06-09 13:09:21 +02:00
• 1dde829e34 UOps.IF* to graph spec (#4894) qazalandGitHub 2024-06-09 19:00:12 +08:00
• b9afb0d577 test uop as symbolic (#4870) George HotzandGitHub 2024-06-09 12:15:11 +02:00
• 654a8b9ef7 retire hsa (#4885) nimlgenandGitHub 2024-06-09 11:33:03 +03:00
• e33efd6a3d test cases for multitensor adds const (#4892) chenyuandGitHub 2024-06-08 22:57:48 -04:00
• a3ec4234df expand broadcast functions a bit (#4891) chenyuandGitHub 2024-06-08 20:16:54 -04:00
• 2849d0a2a1 fix copying to clipboard on a non secure context (#4890) wozeparrotandGitHub 2024-06-08 23:51:47 +00:00
• 6327b50e51 amd in benchmarks (#4861) nimlgenandGitHub 2024-06-08 23:24:46 +03:00
• d24e57c615 amd support kernel with bf16 (#4863) nimlgenandGitHub 2024-06-08 22:52:32 +03:00
• 6c24eda522 feat: tinychat (#4869) wozeparrotandGitHub 2024-06-08 19:05:45 +00:00
• 9445946cae docs: Update referenced yaml in yolov8.py (#4871) Brennan KinneyandGitHub 2024-06-09 07:05:00 +12:00
• 794fecf8e3 perf: faster element deletion during matching (#4882) Roelof van DijkandGitHub 2024-06-08 15:16:35 +02:00
• 0eebb8e998 fix: _free should not return (#4880) Roelof van DijkandGitHub 2024-06-08 14:45:06 +02:00
• 1785a70e77 fix: else-return on runtime (#4881) Roelof van DijkandGitHub 2024-06-08 14:44:24 +02:00
• 1e3325f369 raise assert [run_process_replay] (#4879) qazalandGitHub 2024-06-08 20:31:44 +08:00
• d19f39d4dd unbind Variable pre LazyOp (#4873) qazalandGitHub 2024-06-08 20:16:38 +08:00
• 9c30889ce9 [run_process_replay] faster and simpler match function (#4876) George HotzandGitHub 2024-06-08 14:08:30 +02:00
• aadab3e3da fix: pylint will not lint folders without __init__.py (#4875) Roelof van DijkandGitHub 2024-06-08 14:00:24 +02:00
• 1680a4bcb8 Remove unused and internal variables (#4862) Szymon OżógandGitHub 2024-06-07 23:05:38 +02:00
• 15e5a4fb26 fix: variable defined in assert breaks -O (#4866) Roelof van DijkandGitHub 2024-06-07 20:36:24 +02:00
• 3a20cff7c2 expand ShapeTracker.invert a bit (#4864) chenyuandGitHub 2024-06-07 14:26:02 -04:00
• 688b14c933 do not sleep immediately in amd's wait_signal (#4859) nimlgenandGitHub 2024-06-07 16:33:46 +03:00
• 66dfd5e7bf faster codegen process replay (#4858) qazalandGitHub 2024-06-07 21:20:57 +08:00
• dd5378378b cleanup kernel simplify_merge_adjacent (#4852) chenyuandGitHub 2024-06-06 12:04:54 -04:00
• 47bfd7c2b7 fix sync of offset buffers in graphs (#4850) nimlgenandGitHub 2024-06-06 16:09:45 +03:00
• eeb5a7af39 refactor linearize to render_block, P1 (#4839) qazalandGitHub 2024-06-06 20:31:43 +08:00
• b932ce0f1d [run_process_replay] style: clean up UPat geohot 2024-06-06 08:51:39 +02:00
• b42f49b506 minor cleanup of view _merge_dims (#4849) chenyuandGitHub 2024-06-05 23:20:26 -04:00
• 1649c21ead nv fix round of allocation sizes (#4828) nimlgenandGitHub 2024-06-06 00:21:56 +03:00
• 09bfb8c10a nv sync program copies to other exection (#4845) nimlgenandGitHub 2024-06-05 23:34:33 +03:00
• 99e7a1d5e9 support symbolic reshape with non-contiguous (#4844) chenyuandGitHub 2024-06-05 16:01:19 -04:00
• a352b6d9ce symbolic Tensor.var (#4843) chenyuandGitHub 2024-06-05 12:55:54 -04:00
• 085c0bbf6b add mlperf train subset of openimages (#4841) NikandGitHub 2024-06-05 16:10:11 +02:00
• 887643cf34 Multireduce atomic local load/store test (#4786) 3 people 2024-06-05 04:41:19 -07:00
• 3954f102aa style: make __init__ first in Tensor class geohot 2024-06-05 12:51:41 +02:00
• 273945df67 Regression tests for bitshift (#4829) Szymon OżógandGitHub 2024-06-05 11:42:34 +02:00
• 5ac30c29d8 Construct UOps patterns using UPat (#4821) Alec ChenandGitHub 2024-06-05 03:29:37 -05:00
• e47277d18a Disable for PTX as well (#4838) 3 people 2024-06-05 09:37:59 +02:00
• 890e7c12bb test/external/verify_kernel: add support for single pickled kernel (#4836) Francis LamandGitHub 2024-06-04 15:59:21 -07:00
• e576aca044 Disable dropout (#4837) Elias WahlandGitHub 2024-06-05 00:57:26 +02:00
• bb248a0dd1 Optional half matmul (#4835) Elias WahlandGitHub 2024-06-04 23:53:41 +02:00
• 04e237328b Refactor to class style (#4804) Elias WahlandGitHub 2024-06-04 23:08:31 +02:00
• 1b8bed4a26 nv check cmdq overrun (#4824) nimlgenandGitHub 2024-06-04 23:22:58 +03:00
• cddce0e168 don't cast before view on shape changing bitcast (#4833) David HouandGitHub 2024-06-04 13:04:52 -07:00
• 0c3a996e64 Nest ifs for dtype and uop in pattern matcher (#4834) Alec ChenandGitHub 2024-06-04 14:51:28 -05:00
• 4909a0d16f Fix arg set in pattern matcher (#4830) Alec ChenandGitHub 2024-06-04 14:10:09 -05:00