test_quantize_per_channel gets core dump

[weishi-deng]

🐛 Describe the bug

import torch

def test_quantize_per_channel(dtype=torch.float):
    src_cpu = torch.randn(1, 3, 2, 2)
    src_gpu = src_cpu.to("xpu")

    data_type = torch.qint8
    channel_scale_cpu = torch.Tensor([0.1, 0.3, 0.5])
    channel_zero_point_cpu = torch.tensor([0, 0, 0])
    channel_scale_xpu = torch.Tensor([0.1, 0.3, 0.5]).to("xpu")
    channel_zero_point_xpu = torch.tensor([0, 0, 0]).to("xpu")

    dst_gpu = torch.quantize_per_channel(
        src_gpu,
        scales=channel_scale_xpu,
        zero_points=channel_zero_point_xpu,
        dtype=data_type,
        axis=1,
    )
    print(dst_gpu)

test_quantize_per_channel()

Versions

pytorch 2.7