From 366c9b4526b4b940c8aafce459d6784211e862bc Mon Sep 17 00:00:00 2001
From: Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com>
Date: Tue, 6 Aug 2024 18:24:04 -0400
Subject: Fix auto-diff synthesized method naming conventions (#4714)

* Fix auto-diff synthesized method naming conventions

* Update tests; remove unused var
---
 tests/autodiff/reverse-checkpoint-1.slang | 2 +-
 tests/autodiff/reverse-checkpoint-2.slang | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

(limited to 'tests')

diff --git a/tests/autodiff/reverse-checkpoint-1.slang b/tests/autodiff/reverse-checkpoint-1.slang
index beb983b3b..517297013 100644
--- a/tests/autodiff/reverse-checkpoint-1.slang
+++ b/tests/autodiff/reverse-checkpoint-1.slang
@@ -29,7 +29,7 @@ float f(int p, float x)
 // Check that there are no calls to primal_g in bwd_f.
 
 // CHECK: void s_bwd_f_{{[0-9]+}}
-// CHECK-NOT: {{[_a-zA-Z0-9]+}} = s_bwd_primal_g_{{[0-9]+}}
+// CHECK-NOT: {{[_a-zA-Z0-9]+}} = s_primal_ctx_g_{{[0-9]+}}
 // CHECK: return
 
 
diff --git a/tests/autodiff/reverse-checkpoint-2.slang b/tests/autodiff/reverse-checkpoint-2.slang
index 68ff62176..8a7262aa4 100644
--- a/tests/autodiff/reverse-checkpoint-2.slang
+++ b/tests/autodiff/reverse-checkpoint-2.slang
@@ -29,7 +29,7 @@ float f(int p, float x)
 // Check that there are no calls to primal_g in bwd_f.
 
 // CHECK: void s_bwd_prop_f_{{[0-9]+}}
-// CHECK: {{[_a-zA-Z0-9]+}} = s_bwd_primal_g_{{[0-9]+}}
+// CHECK: {{[_a-zA-Z0-9]+}} = s_primal_ctx_g_{{[0-9]+}}
 // CHECK: return
 
 
-- 
cgit v1.2.3