Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add BF16 in padded FP8 quantize ops #2010

Closed
wants to merge 1 commit into from

Conversation

sryap
Copy link
Contributor

@sryap sryap commented Sep 11, 2023

Summary:

  • Add BF16 support in FloatToPaddedFP8RowwiseQuantized and
    PaddedFP8RowwiseQuantizedToFloat.
  • Refactor src/quantize_ops/quantize_fp8_rowwise.cu
  • Move unit test from hpc to fbgemm_gpu

Differential Revision: D49166595

@facebook-github-bot
Copy link
Contributor

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

@netlify
Copy link

netlify bot commented Sep 11, 2023

Deploy Preview for pytorch-fbgemm-docs canceled.

Name Link
🔨 Latest commit 95a116c
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/650a357d1424980008937bd3

@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Sep 15, 2023
Summary:

- Add BF16 support in `FloatToPaddedFP8RowwiseQuantized` and
  `PaddedFP8RowwiseQuantizedToFloat`.
- Refactor `src/quantize_ops/quantize_fp8_rowwise.cu`
- Move unit test from `hpc` to `fbgemm_gpu`

Reviewed By: summerdengfb, qchip

Differential Revision: D49166595
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Sep 15, 2023
Summary:

- Add BF16 support in `FloatToPaddedFP8RowwiseQuantized` and
  `PaddedFP8RowwiseQuantizedToFloat`.
- Refactor `src/quantize_ops/quantize_fp8_rowwise.cu`
- Move unit test from `hpc` to `fbgemm_gpu`

Reviewed By: summerdengfb, qchip

Differential Revision: D49166595
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Sep 19, 2023
Summary:

- Add BF16 support in `FloatToPaddedFP8RowwiseQuantized` and
  `PaddedFP8RowwiseQuantizedToFloat`.
- Refactor `src/quantize_ops/quantize_fp8_rowwise.cu`
- Move unit test from `hpc` to `fbgemm_gpu`

Reviewed By: summerdengfb, qchip

Differential Revision: D49166595
@facebook-github-bot
Copy link
Contributor

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

1 similar comment
@facebook-github-bot
Copy link
Contributor

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

Summary:

- Add BF16 support in `FloatToPaddedFP8RowwiseQuantized` and
  `PaddedFP8RowwiseQuantizedToFloat`.
- Refactor `src/quantize_ops/quantize_fp8_rowwise.cu`
- Move unit test from `hpc` to `fbgemm_gpu`

Reviewed By: jianyuh, summerdengfb, qchip

Differential Revision: D49166595
@facebook-github-bot
Copy link
Contributor

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

1 similar comment
@facebook-github-bot
Copy link
Contributor

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

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 3950764.

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.

None yet

2 participants