From d5c6e95a85e4ee3e474d344ce4509f6e1f06357e Mon Sep 17 00:00:00 2001
From: "yan.yan" <yanyan.sub@outlook.com>
Date: Sun, 1 Sep 2024 23:54:55 +0800
Subject: [PATCH] fix apple metal problem

---
 include/tensorview/geometry/obb_grid_overlap.h | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/include/tensorview/geometry/obb_grid_overlap.h b/include/tensorview/geometry/obb_grid_overlap.h
index 7d055e9..58bbe16 100644
--- a/include/tensorview/geometry/obb_grid_overlap.h
+++ b/include/tensorview/geometry/obb_grid_overlap.h
@@ -162,10 +162,10 @@ template <typename T> class OBBCornersGridOverlap {
   TV_HOST_DEVICE_INLINE static tv::array_nd<T, 4, 2>
   prepare_dfvt_corners(const TV_METAL_THREAD tv::array_nd<T, 4, 2> &corners,
                        const TV_METAL_THREAD tv::array<T, 2> &major_vector) {
-    auto &p_top_right = corners[0];
-    auto &p_bottom_right = corners[1];
-    auto &p_bottom_left = corners[2];
-    auto &p_top_left = corners[3];
+    TV_METAL_THREAD auto &p_top_right = corners[0];
+    TV_METAL_THREAD auto &p_bottom_right = corners[1];
+    TV_METAL_THREAD auto &p_bottom_left = corners[2];
+    TV_METAL_THREAD auto &p_top_left = corners[3];
     if (major_vector[0] >= 0 && major_vector[1] >= 0) {
       return {(p_bottom_left), (p_top_left), (p_bottom_right), (p_top_right)};
     } else if (major_vector[0] < 0 && major_vector[1] >= 0) {