From d92a09d823477f9a6bb2a6b787aee700950978c8 Mon Sep 17 00:00:00 2001 From: Jee Jee Li Date: Wed, 23 Oct 2024 10:23:56 +0000 Subject: [PATCH] test sign-off Signed-off-by: Jee Jee Li --- csrc/torch_bindings.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/csrc/torch_bindings.cpp b/csrc/torch_bindings.cpp index 645c4009cd3c4..826f918c82e78 100644 --- a/csrc/torch_bindings.cpp +++ b/csrc/torch_bindings.cpp @@ -60,7 +60,7 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) { ops.def("gelu_tanh_and_mul(Tensor! out, Tensor input) -> ()"); ops.impl("gelu_tanh_and_mul", torch::kCUDA, &gelu_tanh_and_mul); - // FATReLU implementation + // FATReLU implementation. ops.def("fatrelu_and_mul(Tensor! out, Tensor input, float threshold) -> ()"); ops.impl("fatrelu_and_mul", torch::kCUDA, &fatrelu_and_mul);