Skip to content

Add Float8_e8m0fnu support for copy - #2258

Merged
CuiYifeng merged 2 commits into
mainfrom
yifeng/fp8_e8m0_copy
Nov 5, 2025
Merged

CuiYifeng merged 2 commits into
mainfrom
yifeng/fp8_e8m0_copy

Conversation

@CuiYifeng

Copy link
Copy Markdown
Contributor

To solve #2207
This PR is to support Float8_e8m0fnu copy on XPU backend.

Copilot AI review requested due to automatic review settings October 31, 2025 06:40
@CuiYifeng CuiYifeng added this to the PT2.10 milestone Oct 31, 2025

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull Request Overview

This PR adds Float8_e8m0fnu support for copy operations on XPU backend to solve issue #2207. The changes enable Float8_e8m0fnu dtype for various tensor operations including copy, flip, and indexing operations.

  • Added Float8_e8m0fnu support to copy kernel with proper casting functions
  • Updated dispatch macros to include all Float8 types using AT_FLOAT8_TYPES
  • Enabled Float8_e8m0fnu in test coverage by removing skip list entries

Reviewed Changes

Copilot reviewed 7 out of 7 changed files in this pull request and generated 2 comments.

Show a summary per file
File Description
test/xpu/skip_list_common.py Removed Float8 skip for indexing tests
test/regressions/test_index_and_index_put.py Added Float8 flip test including Float8_e8m0fnu
test/regressions/test_copy.py Added Float8_e8m0fnu test coverage for copy operations
src/ATen/native/xpu/sycl/TensorTransformationsKernels.cpp Updated flip kernel to support Float8 types
src/ATen/native/xpu/sycl/Indexing.cpp Updated indexing kernels to support Float8 types
src/ATen/native/xpu/sycl/CopyKernel.cpp Added Float8_e8m0fnu copy kernel implementation
src/ATen/native/xpu/XPUScalar.cpp Added Float8 types to scalar conversion

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Comment thread test/regressions/test_copy.py Resolved
Comment thread src/ATen/native/xpu/sycl/Indexing.cpp Resolved
@CuiYifeng

Copy link
Copy Markdown
Contributor Author

Depends on #2190.

@CuiYifeng
CuiYifeng force-pushed the yifeng/fp8_e8m0_copy branch from c525192 to aeeb7f7 Compare November 3, 2025 02:21
@CuiYifeng
CuiYifeng added this pull request to the merge queue Nov 5, 2025
Merged via the queue into main with commit 58b063e Nov 5, 2025
96 of 100 checks passed
@CuiYifeng
CuiYifeng deleted the yifeng/fp8_e8m0_copy branch November 5, 2025 10:13
mengfei25 added a commit that referenced this pull request Nov 7, 2025
To solve #2207
This PR is to support Float8_e8m0fnu copy on XPU backend.

Co-authored-by: mengfei25 <mengfei.li@Intel.com>
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.

5 participants