Mismatch in kernel C++ signatures
operator: aten::rrelu_with_noise(Tensor self, Tensor(b!) noise, Scalar lower=0.125, Scalar upper=0.33333333333333331, bool training=False, Generator? generator=Non
e) -> Tensor
registered at /home/xinanlin/xinanlin/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
kernel 1: at::Tensor (at::Tensor const&, at::Tensor&, c10::Scalar const&, c10::Scalar const&, bool, std::optional<at::Generator>)
dispatch key: FuncTorchBatched
registered at /home/xinanlin/xinanlin/pytorch/aten/src/ATen/functorch/BatchRulesBinaryOps.cpp:353
kernel 2: at::Tensor (at::Tensor const&, at::Tensor const&, c10::Scalar const&, c10::Scalar const&, bool, std::optional<at::Generator>)
dispatch key: XPU
registered at /home/xinanlin/xinanlin/pytorch/build/xpu/ATen/RegisterXPU.cpp:39552
Exception raised from registerKernel at /home/xinanlin/xinanlin/pytorch/aten/src/ATen/core/dispatch/OperatorEntry.cpp:121 (most recent call first):
frame #0: c10::Error::Error(c10::SourceLocation, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >) + 0xac (0x7f871fbfec0c in /home/xin
anlin/xinanlin/pytorch/torch/lib/libc10.so)
frame #1: c10::detail::torchCheckFail(char const*, char const*, unsigned int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > const&)
+ 0xf3 (0x7f871fba2df7 in /home/xinanlin/xinanlin/pytorch/torch/lib/libc10.so)
frame #2: c10::impl::OperatorEntry::registerKernel(c10::Dispatcher const&, std::optional<c10::DispatchKey>, c10::KernelFunction, std::optional<c10::impl::CppSignatur
e>, std::unique_ptr<c10::FunctionSchema, std::default_delete<c10::FunctionSchema> >, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >)
+ 0x28e (0x7f87263f30ce in /home/xinanlin/xinanlin/pytorch/torch/lib/libtorch_cpu.so)
frame #3: c10::Dispatcher::registerImpl(c10::OperatorName, std::optional<c10::DispatchKey>, c10::KernelFunction, std::optional<c10::impl::CppSignature>, std::unique_
ptr<c10::FunctionSchema, std::default_delete<c10::FunctionSchema> >, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >) + 0x15c (0x7f87
263e6a1c in /home/xinanlin/xinanlin/pytorch/torch/lib/libtorch_cpu.so)
frame #4: torch::Library::_impl(char const*, torch::CppFunction&&, torch::_RegisterOrVerify) & + 0x3b6 (0x7f87264275b6 in /home/xinanlin/xinanlin/pytorch/torch/lib/l
ibtorch_cpu.so)
frame #5: <unknown function> + 0x5ce19e (0x7f87205f319e in /home/xinanlin/xinanlin/pytorch/torch/lib/libtorch_xpu.so)
frame #6: <unknown function> + 0x5deb2b (0x7f8720603b2b in /home/xinanlin/xinanlin/pytorch/torch/lib/libtorch_xpu.so)
frame #7: <unknown function> + 0x60fa2d (0x7f8720634a2d in /home/xinanlin/xinanlin/pytorch/torch/lib/libtorch_xpu.so)
frame #8: <unknown function> + 0x3c89ab (0x7f87203ed9ab in /home/xinanlin/xinanlin/pytorch/torch/lib/libtorch_xpu.so)
Python version: 3.9.20 | packaged by conda-forge | (main, Sep 30 2024, 17:49:10) [GCC 13.3.0] (64-bit runtime)
Python platform: Linux-5.15.47+prerelease23.5.18-x86_64-with-glibc2.35
🐛 Describe the bug
The PR #141867 has changed the C++ signatures of aten::rrelu_with_noise, but torch-xpu-ops is not aligned, and cause the following error:
Versions
PyTorch version: 2.6.0a0+gitf85e23
Is debug build: False
CUDA used to build PyTorch: None
ROCM used to build PyTorch: N/A
OS: Ubuntu 22.04.5 LTS (x86_64)
GCC version: (Ubuntu 11.4.0-1ubuntu1~22.04) 11.4.0
Clang version: Could not collect
CMake version: version 3.29.4
Libc version: glibc-2.35
Python version: 3.9.20 | packaged by conda-forge | (main, Sep 30 2024, 17:49:10) [GCC 13.3.0] (64-bit runtime)
Python platform: Linux-5.15.47+prerelease23.5.18-x86_64-with-glibc2.35
cc @gujinghui @EikanWang @fengyuan14 @guangyey