getnv(a,...) -> getnv(b,...)

Lightning-AI · Dec 21, 2024 · fce281b · fce281b
1 parent 6a6fe59
commit fce281b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thunder/executors/nvfuserex_impl.py b/thunder/executors/nvfuserex_impl.py
@@ -2659,7 +2659,7 @@ def _nv_scaled_mm(
     lc_to_nv_map: dict,
 ) -> Any:
     nv_a = getnv(a, fd, lc_to_nv_map)
-    nv_b = getnv(a, fd, lc_to_nv_map)
+    nv_b = getnv(b, fd, lc_to_nv_map)
     nvscale_a = getnv(scale_a, fd, lc_to_nv_map)
     nvscale_b = getnv(scale_b, fd, lc_to_nv_map)