From 065bd31af5f3847f36e448638c88d26c4c82bd2c Mon Sep 17 00:00:00 2001
From: Damian Szwichtenberg <damian.szwichtenberg@intel.com>
Date: Mon, 20 Nov 2023 13:41:29 +0100
Subject: [PATCH] Update pyg_lib/ops/__init__.py

Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>
---
 pyg_lib/ops/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyg_lib/ops/__init__.py b/pyg_lib/ops/__init__.py
index 9c2c2a78a..113e9bc80 100644
--- a/pyg_lib/ops/__init__.py
+++ b/pyg_lib/ops/__init__.py
@@ -361,7 +361,7 @@ def softmax_csr(
                 [0.7792, 0.3502, 0.1638, 0.2145]])
     """
     dim = dim + src.dim() if dim < 0 else dim
-    return torch.ops.pyg.softmax_csr_forward(src, ptr, dim)
+    return torch.ops.pyg.softmax_csr(src, ptr, dim)
 
 
 __all__ = [