From 30fa5843aeb1ac48816967a63db140cff6044e13 Mon Sep 17 00:00:00 2001
From: Zihao Ye <expye@outlook.com>
Date: Wed, 6 Mar 2024 18:14:35 -0800
Subject: [PATCH] fix: fix FindThrust.cmake (#161)

The latest thrust version corresponding to CUDA 12.4 is 200301 which is
not divisible by 100.
Use predefined `THRUST_MINOR_VERSION`/ `THRUST_MAJOR_VERSION` instead.
---
 cmake/modules/FindThrust.cmake | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

diff --git a/cmake/modules/FindThrust.cmake b/cmake/modules/FindThrust.cmake
index 5f38d416..a0f8008f 100644
--- a/cmake/modules/FindThrust.cmake
+++ b/cmake/modules/FindThrust.cmake
@@ -51,13 +51,32 @@ if( THRUST_INCLUDE_DIR )
     version
     "${version}"
     )
+
+  file( STRINGS ${THRUST_INCLUDE_DIR}/thrust/version.h
+    major_version
+    REGEX "#define THRUST_MAJOR_VERSION[ \t]+([0-9x]+)"
+    )
+  string( REGEX REPLACE
+    "#define THRUST_MAJOR_VERSION[ \t]+"
+    ""
+    major_version
+    "${major_version}"
+    )
+
+  file( STRINGS ${THRUST_INCLUDE_DIR}/thrust/version.h
+    major_version
+    REGEX "#define THRUST_MINOR_VERSION[ \t]+([0-9x]+)"
+    )
+  string( REGEX REPLACE
+    "#define THRUST_MINOR_VERSION[ \t]+"
+    ""
+    minor_version
+    "${minor_version}"
+    )
   
-  math(EXPR major "${version} / 100000")
-  math(EXPR minor "(${version} / 100) % 1000")
-  math(EXPR version "${version} % 100")
-  set( THRUST_VERSION "${major}.${minor}.${version}")
-  set( THRUST_MAJOR_VERSION "${major}")
-  set( THRUST_MINOR_VERSION "${minor}")
+  set( THRUST_VERSION "${version}")
+  set( THRUST_MAJOR_VERSION "${major_version}")
+  set( THRUST_MINOR_VERSION "${minor_version}")
 endif( THRUST_INCLUDE_DIR )
 
 # Check for required components