Add building gtest for OpenCL in GPU build

apache · Nov 23, 2022 · e81ae5e · e81ae5e
1 parent d71de34
commit e81ae5e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/tests/cpp-runtime/opencl/opencl_texture_pool_test.cc b/tests/cpp-runtime/opencl/opencl_texture_pool_test.cc
@@ -31,13 +31,13 @@ using namespace tvm::runtime::cl;
 // get and check internal state of class Pool
 class PoolWrapper : public Pool2D {
  public:
-  inline size_t FreeListSize() const { return free_list_.size(); }
-  inline size_t AllocatedListSize() const { return allocated_.size(); }
-  inline std::pair<size_t, size_t> FreeListItemSize(size_t idx) const {
-    return std::make_pair(free_list_[idx].x, free_list_[idx].y);
+  inline int FreeListSize() const { return static_cast<int>(free_list_.size()); }
+  inline int AllocatedListSize() const { return static_cast<int>(allocated_.size()); }
+  inline std::pair<int, int> FreeListItemSize(size_t idx) const {
+    return std::make_pair(static_cast<int>(free_list_[idx].x), static_cast<int>(free_list_[idx].y));
   }
-  inline std::pair<size_t, size_t> AllocatedListItemSize(size_t idx) const {
-    return std::make_pair(allocated_[idx].x, allocated_[idx].y);
+  inline std::pair<int, int> AllocatedListItemSize(size_t idx) const {
+    return std::make_pair(static_cast<int>(allocated_[idx].x), static_cast<int>(allocated_[idx].y));
   }
 };
 

diff --git a/tests/scripts/task_config_build_gpu.sh b/tests/scripts/task_config_build_gpu.sh
@@ -29,6 +29,7 @@ echo set\(USE_CUDA ON\) >> config.cmake
 echo set\(USE_VULKAN ON\) >> config.cmake
 echo set\(USE_OPENGL ON\) >> config.cmake
 echo set\(USE_OPENCL ON\) >> config.cmake
+echo set\(USE_OPENCL_GTEST /googletest\) >> config.cmake
 echo set\(USE_MICRO_STANDALONE_RUNTIME ON\) >> config.cmake
 echo set\(USE_LLVM \"/usr/bin/llvm-config-9 --link-static\"\) >> config.cmake
 echo set\(USE_NNPACK ON\) >> config.cmake