update

taichi-dev · Jan 4, 2023 · f5b0dc6 · f5b0dc6
1 parent b5c4588
commit f5b0dc6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/taichi/rhi/amdgpu/amdgpu_context.h b/taichi/rhi/amdgpu/amdgpu_context.h
@@ -34,6 +34,11 @@ class AMDGPUContext {
     return dev_count_ != 0;
   }
 
+  void pack_args(std::vector<void *> arg_pointers,
+                 std::vector<int> arg_sizes, char *arg_packed);
+
+  int get_args_byte(std::vector<int> arg_sizes)
+
   void launch(void *func,
               const std::string &task_name,
               const std::vector<void *> &arg_pointers,