build(llama.cu): 兼容天数

Signed-off-by: YdrMaster <ydrml@hotmail.com>

Author: YdrMaster

Cargo.lock

@@ -4,7 +4,7 @@ version = "0.0.0"
 edition.workspace = true
 
 [dependencies]
-operators = { git = "https://github.com/YdrMaster/operators-rs", rev = "ed28f2d", default-features = false, features = [
+operators = { git = "https://github.com/YdrMaster/operators-rs", rev = "01b39e8", default-features = false, features = [
     "nvidia-gpu",
 ] }
 nn = { git = "https://github.com/YdrMaster/InfiniNN", rev = "1fff7ec" }
@@ -27,5 +27,6 @@ minijinja = { version = "2.10", default-features = false, features = [
 [build-dependencies]
 bindgen = { version = "0.71" }
 build-script-cfg = "0.1"
-search-cuda-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "106f0b6" }
-search-maca-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "106f0b6" }
+search-cuda-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "b0148c0" }
+search-maca-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "b0148c0" }
+search-corex-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "b0148c0" }

llama.cu/Cargo.toml

@@ -1,4 +1,5 @@
 use build_script_cfg::Cfg;
+use search_corex_tools::find_corex;
 use search_cuda_tools::{find_cuda_root, find_nccl_root};
 use search_maca_tools::find_maca_root;
 use std::{
@@ -18,6 +19,14 @@ fn main() {
             ["-x", "hpcc", "-fPIC"],
             Some("__MACA_ARCH__"),
         )
+    } else if let Some(corex_root) = find_corex() {
+        compile_bind(
+            corex_root,
+            "sample.cu",
+            "bin/clang++",
+            ["-x", "ivcore", "-fPIC"],
+            None,
+        )
     } else if let Some(cuda_root) = find_cuda_root() {
         if find_nccl_root().is_some() {
             nccl.define()