[llvm] [aot] Add LLVM to CAPI part 5: Added C-API tests for Vulkan and Cuda backend

c_api/src/c_api_test_utils.cpp

@@ -0,0 +1,24 @@
+#include "c_api_test_utils.h"
+#include "taichi/platform/cuda/detect_cuda.h"
+
+#ifdef TI_WITH_VULKAN
+#include "taichi/rhi/vulkan/vulkan_loader.h"
+#endif
+
+namespace capi {
+namespace utils {
+
+bool is_vulkan_available() {
+#ifdef TI_WITH_VULKAN
+  return taichi::lang::vulkan::is_vulkan_api_available();
+#else
+  return false;
+#endif
+}
+
+bool is_cuda_available() {
+  return taichi::is_cuda_api_available();
+}
+
+}  // namespace utils
+}  // namespace capi

c_api/src/c_api_test_utils.h

@@ -0,0 +1,11 @@
+#pragma once
+#include <taichi/taichi_platform.h>
+
+namespace capi {
+namespace utils {
+
+TI_DLL_EXPORT bool TI_API_CALL is_vulkan_available();
+TI_DLL_EXPORT bool TI_API_CALL is_cuda_available();
+
+}  // namespace utils
+}  // namespace capi

c_api/tests/c_api_interface_test.cpp

@@ -1,4 +1,5 @@
 #include "gtest/gtest.h"
+#include "c_api_test_utils.h"
 #include "taichi/taichi_core.h"
 
 TEST(CapiDryRun, Runtime) {
@@ -8,6 +9,20 @@ TEST(CapiDryRun, Runtime) {
     TiRuntime runtime = ti_create_runtime(arch);
     ti_destroy_runtime(runtime);
   }
+
+  if (capi::utils::is_vulkan_available()) {
+    // Vulkan Runtime
+    TiArch arch = TiArch::TI_ARCH_VULKAN;
+    TiRuntime runtime = ti_create_runtime(arch);
+    ti_destroy_runtime(runtime);
+  }
+
+  if (capi::utils::is_cuda_available()) {
+    // Vulkan Runtime
+    TiArch arch = TiArch::TI_ARCH_CUDA;
+    TiRuntime runtime = ti_create_runtime(arch);
+    ti_destroy_runtime(runtime);
+  }
 }
 
 TEST(CapiDryRun, MemoryAllocation) {
@@ -30,6 +45,28 @@ TEST(CapiDryRun, MemoryAllocation) {
 
     ti_destroy_runtime(runtime);
   }
+
+  if (capi::utils::is_vulkan_available()) {
+    // Vulkan Runtime
+    TiArch arch = TiArch::TI_ARCH_VULKAN;
+    TiRuntime runtime = ti_create_runtime(arch);
+
+    TiMemory memory = ti_allocate_memory(runtime, &alloc_info);
+    ti_free_memory(runtime, memory);
+
+    ti_destroy_runtime(runtime);
+  }
+
+  if (capi::utils::is_cuda_available()) {
+    // Cuda Runtime
+    TiArch arch = TiArch::TI_ARCH_CUDA;
+    TiRuntime runtime = ti_create_runtime(arch);
+
+    TiMemory memory = ti_allocate_memory(runtime, &alloc_info);
+    ti_free_memory(runtime, memory);
+
+    ti_destroy_runtime(runtime);
+  }
 }
 
 TEST(CapiDryRun, CpuAotModule) {
@@ -49,3 +86,45 @@ TEST(CapiDryRun, CpuAotModule) {
     ti_destroy_runtime(runtime);
   }
 }
+
+TEST(CapiDryRun, VulkanAotModule) {
+  if (capi::utils::is_vulkan_available()) {
+    const auto folder_dir = getenv("TAICHI_AOT_FOLDER_PATH");
+
+    std::stringstream aot_mod_ss;
+    aot_mod_ss << folder_dir;
+
+    {
+      // Vulkan Runtime
+      TiArch arch = TiArch::TI_ARCH_VULKAN;
+      TiRuntime runtime = ti_create_runtime(arch);
+
+      TiAotModule aot_mod =
+          ti_load_aot_module(runtime, aot_mod_ss.str().c_str());
+      ti_destroy_aot_module(aot_mod);
+
+      ti_destroy_runtime(runtime);
+    }
+  }
+}
+
+TEST(CapiDryRun, CudaAotModule) {
+  if (capi::utils::is_cuda_available()) {
+    const auto folder_dir = getenv("TAICHI_AOT_FOLDER_PATH");
+
+    std::stringstream aot_mod_ss;
+    aot_mod_ss << folder_dir;
+
+    {
+      // Vulkan Runtime
+      TiArch arch = TiArch::TI_ARCH_CUDA;
+      TiRuntime runtime = ti_create_runtime(arch);
+
+      TiAotModule aot_mod =
+          ti_load_aot_module(runtime, aot_mod_ss.str().c_str());
+      ti_destroy_aot_module(aot_mod);
+
+      ti_destroy_runtime(runtime);
+    }
+  }
+}

cmake/TaichiCAPI.cmake

@@ -3,6 +3,10 @@ cmake_minimum_required(VERSION 3.0)
 set(TAICHI_C_API_NAME taichi_c_api)
 
 file(GLOB_RECURSE C_API_SOURCE "c_api/src/*.cpp")
+if(NOT TI_BUILD_TESTS)
+    list(REMOVE_ITEM C_API_SOURCE "c_api/src/c_api_test_utils.cpp")
+endif()
+
 add_library(${TAICHI_C_API_NAME} SHARED ${C_API_SOURCE})
 target_link_libraries(${TAICHI_C_API_NAME} PRIVATE taichi_isolated_core)
 

cmake/TaichiCAPITests.cmake

@@ -22,6 +22,7 @@ target_include_directories(${C_API_TESTS_NAME}
   PRIVATE
     ${PROJECT_SOURCE_DIR}
     ${PROJECT_SOURCE_DIR}/c_api/include
+    ${PROJECT_SOURCE_DIR}/c_api/src
   )
 
 add_test(NAME ${C_API_TESTS_NAME} COMMAND ${C_API_TESTS_NAME})

tests/cpp/aot/llvm/kernel_aot_test.py

@@ -13,5 +13,7 @@
         compile_kernel_aot(arch=ti.cpu)
     elif args.arch == "cuda":
         compile_kernel_aot(arch=ti.cuda)
+    elif args.arch == "vulkan":
+        compile_kernel_aot(arch=ti.vulkan)
     else:
         assert False

tests/test_utils.py

@@ -36,6 +36,12 @@
 __capi_aot_test_cases = {
     "CapiDryRun.CpuAotModule":
     [os.path.join('cpp', 'aot', 'llvm', 'kernel_aot_test.py'), "--arch=cpu"],
+    "CapiDryRun.VulkanAotModule": [
+        os.path.join('cpp', 'aot', 'llvm', 'kernel_aot_test.py'),
+        "--arch=vulkan"
+    ],
+    "CapiDryRun.CudaAotModule":
+    [os.path.join('cpp', 'aot', 'llvm', 'kernel_aot_test.py'), "--arch=cuda"],
 }