Skip to content

Chunk expensive numerical operations - #1025

Merged
ValterH merged 1 commit into
mainfrom
mem-opt/chunk-numerical-ops-main
Sep 29, 2026
Merged

ValterH merged 1 commit into
mainfrom
mem-opt/chunk-numerical-ops-main

Conversation

@ValterH

@ValterH ValterH commented Sep 29, 2026

Copy link
Copy Markdown
Collaborator

Requested by @ValterH: “create a new branch and properly merge 1016 into main.”

AI-generated draft summary: Bring #1016 onto main; the original PR was merged into mem-opt/base-numerical-chunks.

Add split_size and use it to compute RobustScale quantiles over column chunks and ClipSigma transforms over row chunks. Preserve the non-chunked autograd path and include chunk-equivalence checks.

Apply only the chunking commit on top of current main, retaining the final #1014 implementation without restoring its removed fitting-gradient test.

Co-authored-by: Jingang Qu <jqu@nvidia.com>
Co-authored-by: Cedric Lorenz <clorenz@nvidia.com>
@copy-pr-bot

copy-pr-bot Bot commented Sep 29, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@ValterH
ValterH marked this pull request as ready for review September 29, 2026 12:38
@ValterH
ValterH merged commit 10829fd into main Sep 29, 2026
3 of 4 checks passed
@ValterH
ValterH deleted the mem-opt/chunk-numerical-ops-main branch September 29, 2026 12:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants