Skip to content

Pull requests: google/jaxite

New pull request New
8 Open 43 Closed
8 Open 43 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

The first trail to integrate BAT into CGGI Bootstrapping with BMMP (online compilation).
#53 opened Dec 13, 2024 by copybara-service bot Loading…
1
increase vmem_limit_bytes to 32MiB to fit the 18.8 MiB on-chip VMEM requirement.
#52 opened Dec 13, 2024 by copybara-service bot Loading…
1
We now have two versions "_i32_matmul_unreduced_matmul" one comes with unique optimization for CGGI Bootstrapping and does not work for the original polymul_kernel_test/test_i32_matmul_vs_refernce. So add back the original version for this test case.
#51 opened Dec 12, 2024 by copybara-service bot Loading…
1
Merge two (18, 3, 512) into a single (18, 8, 512), further reducing latency from 7643 us to 7452 us.
#50 opened Dec 11, 2024 by copybara-service bot Loading…
1
Enable BMMP as on/off switch "use_bmmp".
#49 opened Dec 11, 2024 by copybara-service bot Loading…
1
Update references to JAX's GitHub repo
#46 opened Dec 5, 2024 by copybara-service bot Loading…
Formatted and Verified High-Performance MatMul.
#45 opened Oct 9, 2024 by copybara-service bot Loading…
Update SR_onboarding tutorial
#43 opened Oct 9, 2024 by copybara-service bot Loading…
1
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.