Skip to content

Add rounding mode options to mx_github#2933

Closed
summerdengfb wants to merge 1 commit into
pytorch:mainfrom
summerdengfb:export-D60682029
Closed

Add rounding mode options to mx_github#2933
summerdengfb wants to merge 1 commit into
pytorch:mainfrom
summerdengfb:export-D60682029

Conversation

@summerdengfb

Copy link
Copy Markdown
Contributor

Summary: We found the rounding mode when selecting the block scales in MX formats is critical to numeric accuracies. Josh has added rounding mode options in the fp32 to mx4 conversion kernels in D59562029. Here is to add the options to mx_github library which supports more mx formats. To use different options from the default 'floor' method in the cuda kernel, need to specify custom_cuda=False in the mx specs.

Differential Revision: D60682029

Summary: We found the rounding mode when selecting the block scales in MX formats is critical to numeric accuracies. Josh has added rounding mode options in the fp32 to mx4 conversion kernels in D59562029. Here is to add the options to mx_github library which supports more mx formats. To use different options from the default 'floor' method in the cuda kernel,  need to specify custom_cuda=False in the mx specs.

Differential Revision: D60682029
@netlify

netlify Bot commented Aug 2, 2024

Copy link
Copy Markdown

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit 8eebf64
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66ad60c1660d400008d93574
😎 Deploy Preview https://deploy-preview-2933--pytorch-fbgemm-docs.netlify.app
📱 Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify site configuration.

@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request was exported from Phabricator. Differential Revision: D60682029

@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request has been merged in b0c16d7.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants