Skip to content

Commit faa79b7

Browse files
committed
Modified some review comments
1 parent 5a08d2e commit faa79b7

3 files changed

Lines changed: 7 additions & 2 deletions

File tree

‎src/ATen/native/xpu/LinearInt4.cpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
#include <comm/xpu_aten.h>
99

1010
namespace at::native {
11-
Tensor& linear_int4_xpu(
11+
Tensor& _weight_int4pack_mm_with_scales_and_zeros_xpu(
1212
const Tensor& input,
1313
const Tensor& weight,
1414
int qGroupSize,

‎src/ATen/native/xpu/sycl/LinearInt4.cpp‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,11 @@ struct LinearInt4KernelFunctor : public __SYCL_KER_CONFIG_CONVENTION__ {
4747
auto bptr = B + g_n * k / 2;
4848
auto aptr = A;
4949
auto cptr = C + g_n;
50+
51+
// AT_DISPATCH_ALL_TYPES_AND_COMPLEX_AND2
52+
// (kBFloat16, kBFloat16, scalar_t, "LinearInt4KernelFunctor", [&] {
53+
54+
5055
if constexpr (std::is_same_v<scalar_t, sycl::half>) {
5156
sycl::half2 tmpAcc = {0.f, 0.f};
5257
uint8_t tmps8[TileK / 2];

‎yaml/native/native_functions.yaml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8549,7 +8549,7 @@
85498549

85508550
- func: _weight_int4pack_mm_with_scales_and_zeros(Tensor self, Tensor mat2, int qGroupSize, Tensor qScaleAndZeros) -> Tensor
85518551
dispatch:
8552-
XPU: linear_int4_xpu
8552+
XPU: _weight_int4pack_mm_with_scales_and_zeros_xpu
85538553
autogen: _weight_int4pack_mm_with_scales_and_zeros.out
85548554
tags: core
85558555

0 commit comments

Comments
 (0)