Test CICD 70

SeisSol · Jul 2, 2024 · 41ba6b4 · 41ba6b4
1 parent 68c1570
commit 41ba6b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorforge/codegen/visitor.py b/tensorforge/codegen/visitor.py
@@ -471,7 +471,7 @@ def generate(self, cpp, namespace, testName, kernelClass, cfg, gemm_cfg, testFra
     with cpp.Function(**testFramework.functionArgs(testName)):
 
       if self._target == 'gpu':
-        cpp('numElements = 1;')
+        cpp(f'int {BatchedOperationsAux.NUM_ELEMENTS_NAME} = 1;')
         cpp(f'yateto::LinearAllocatorT<{self._arch.typename}> linearAllocator;')
 
       factory = UnitTestFactory(cpp, self._arch, self._name, testFramework, self._target)