fix torch fn

NVIDIA · Jan 16, 2025 · a4a8e33 · a4a8e33
1 parent e5b0594
commit a4a8e33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/test_embedding.py b/tests/python/test_embedding.py
@@ -73,5 +73,5 @@ def fusion_func(
     norm_type = 2.0 if norm_type is None else norm_type
     scale_grad_by_freq = False if scale_grad_by_freq is None else scale_grad_by_freq
     sparse = False if sparse is None else sparse
-    ref_out = F.embedding_fwd(input, weight, padding_idx, max_norm, norm_type, scale_grad_by_freq, sparse)
+    ref_out = F.embedding(input, weight, padding_idx, max_norm, norm_type, scale_grad_by_freq, sparse)
     torch.testing.assert_close(nvf_out[0], ref_out)