[refactor] Exchange compiled_grad_functions and compiled_functions in…

… kernel_impl.py (taichi-dev#2930)
larryteal · Sep 14, 2021 · fc22fd4 · fc22fd4
1 parent 96397e7
commit fc22fd4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/taichi/lang/kernel_impl.py b/python/taichi/lang/kernel_impl.py
@@ -325,9 +325,9 @@ def __init__(self, func, is_grad, classkernel=False):
     def reset(self):
         self.runtime = impl.get_runtime()
         if self.is_grad:
-            self.compiled_functions = self.runtime.compiled_functions
-        else:
             self.compiled_functions = self.runtime.compiled_grad_functions
+        else:
+            self.compiled_functions = self.runtime.compiled_functions
 
     def extract_arguments(self):
         sig = inspect.signature(self.func)