Skip to content

Commit

Permalink
[PyTorch] Fix CP implementation with FP8 (#1483)
Browse files Browse the repository at this point in the history
* commit some debug code

Signed-off-by: Xiaowei Ren <[email protected]>

* add more debug info

Signed-off-by: Xiaowei Ren <[email protected]>

* debug code commit and typo fix

Signed-off-by: Xiaowei Ren <[email protected]>

* a typo fix

Signed-off-by: Xiaowei Ren <[email protected]>

* remove debug info

Signed-off-by: Xiaowei Ren <[email protected]>

* do not return lse

Signed-off-by: Xiaowei Ren <[email protected]>

* add amax_per_step for quantizers of CP

Signed-off-by: Xiaowei Ren <[email protected]>

* fix FP8 + CP

Signed-off-by: Xiaowei Ren <[email protected]>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* bug fix

Signed-off-by: Xiaowei Ren <[email protected]>

* bug fix

Signed-off-by: Xiaowei Ren <[email protected]>

* dtype fix

Signed-off-by: Xiaowei Ren <[email protected]>

* bug fix

Signed-off-by: Xiaowei Ren <[email protected]>

---------

Signed-off-by: Xiaowei Ren <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Xiaowei Ren <[email protected]>
  • Loading branch information
3 people authored Feb 20, 2025
1 parent b612cde commit 257345a
Show file tree
Hide file tree
Showing 2 changed files with 166 additions and 98 deletions.
Loading

0 comments on commit 257345a

Please sign in to comment.