Skip to content

Working INIT_SRC0_SHMEM and MUL_ACC for legacy and k quantized types#18

Closed
neha-ha wants to merge 6 commits intomasterfrom
k_quants
Closed

Working INIT_SRC0_SHMEM and MUL_ACC for legacy and k quantized types#18
neha-ha wants to merge 6 commits intomasterfrom
k_quants

Conversation

@neha-ha
Copy link
Collaborator

@neha-ha neha-ha commented Feb 11, 2026

fixed thread memory issues-- passing all tests, and model output is coherent.

neha-ha and others added 5 commits February 10, 2026 19:27
* scale jit working

* preliminary working jit for getrows and mulmat, needs refining

* simplified mul_mat preprocessing switch statement

* get_rows fixes, mul_mat refinement

* formatted + last edits

* removed some extraneous prints

* fixed get_rows, fixed workgroup dispatch in mul_mat. no gibberish

* small fix

* some changes, working

* get_rows and mul_mat jit fixed and working

* Update formatting

* formatting

* Add header

---------

Co-authored-by: Neha Abbas <nehaabbas@ReeseLevines-MacBook-Pro.local>
Co-authored-by: Reese Levine <reeselevine1@gmail.com>
@reeselevine
Copy link
Owner

merged in #20

@reeselevine reeselevine closed this Mar 3, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants