From 801f94984dc9bb078ef6ea9487170fd504f0cd83 Mon Sep 17 00:00:00 2001 From: IvanKobzarev Date: Thu, 24 Oct 2024 11:08:03 -0700 Subject: [PATCH] Align rrelu_with_noise schema to reflect noise mutation --- .torch_pin | 1 + torch_xla/csrc/aten_xla_type.cpp | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) create mode 100644 .torch_pin diff --git a/.torch_pin b/.torch_pin new file mode 100644 index 00000000000..2a5ee5a8de3 --- /dev/null +++ b/.torch_pin @@ -0,0 +1 @@ +9ee5e5e563d70244e9317c1c11a4b1ec98d00ed5 diff --git a/torch_xla/csrc/aten_xla_type.cpp b/torch_xla/csrc/aten_xla_type.cpp index 6e98726063f..e308263e4a5 100644 --- a/torch_xla/csrc/aten_xla_type.cpp +++ b/torch_xla/csrc/aten_xla_type.cpp @@ -3019,7 +3019,7 @@ at::Tensor XLANativeFunctions::roll(const at::Tensor& self, } at::Tensor XLANativeFunctions::rrelu_with_noise( - const at::Tensor& self, const at::Tensor& noise, const at::Scalar& lower, + const at::Tensor& self, at::Tensor& noise, const at::Scalar& lower, const at::Scalar& upper, bool training, std::optional generator) { TORCH_LAZY_FN_COUNTER_TIMED_TRACING("xla::");