-
Notifications
You must be signed in to change notification settings - Fork 61
Open
Milestone
Description
🚀 The feature, motivation and pitch
Plan to enable the following ops for FP8/MXFP8:
- _scaled_mm [XPU] [1/2] add fp8 scaled_mm implementation for XPU pytorch/pytorch#165978
- _scaled_grouped_mm
- fill/fill_
- flip/fliplr/flipud Support FP8 in op flip, index_put, and index.Tensor #2190
- index_put/index_put_ Support FP8 in op flip, index_put, and index.Tensor #2190
- index.Tenor/index.Tensor_out Support FP8 in op flip, index_put, and index.Tensor #2190
- index_select/index_select.out
- gather/gather.out
- cat/cat.out Enable FP8 concat_xpu and where_xpu #2152
- eq/eq_
- ne/ne_
- where Enable FP8 concat_xpu and where_xpu #2152
- empty/zeros/ones
- to
- copy (Float8_e8m0fnu)
- clone
- add/sub/mul/div Enable FP8 add/sub/mul/div #2145
- compare Enable FP8 compare_xpu #2154
- normal
Alternatives
No response
Additional context
No response