From 33d6a1abdc7ddc4e8b59432d6f3fdd853279a151 Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Fri, 20 Dec 2024 18:35:34 +0900 Subject: [PATCH] getnv(a,...) -> getnv(b,...) --- thunder/executors/nvfuserex_impl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/thunder/executors/nvfuserex_impl.py b/thunder/executors/nvfuserex_impl.py index 6c9ff7dc71..d62193bb97 100644 --- 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)