fix typo

taichi-dev · Jan 1, 2023 · 45dff60 · 45dff60
1 parent c168a6e
commit 45dff60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/taichi/jit/jit_module.h b/taichi/jit/jit_module.h
@@ -82,7 +82,7 @@ class JITModule {
 #endif
       }
       else if (module_arch() == Arch::amdgpu) {
-#if defined(TI_WITH_CUDA)
+#if defined(TI_WITH_AMDGPU)
         auto arg_bytes = JITModule::get_args_bytes(args...);
         char packed_args[arg_bytes];
         JITModule::init_args_pointers(packed_args, args...);