feat: add load and alloc operation

Author: lucarlig

descend_derive/Cargo.toml

@@ -9,6 +9,8 @@ edition = "2018"
 [lib]
 proc-macro = true
 
+[build-dependencies]
+
 [dependencies]
 syn = { version = "1.0", features = ["full"] }
 quote = "1.0"

descend_derive/build.rs

@@ -0,0 +1,23 @@
+use std::fs;
+use std::path::Path;
+
+fn main() {
+    // Tell Cargo to re-run this build script if any .desc files change
+    let examples_dir = "examples/core";
+    
+    if Path::new(examples_dir).exists() {
+        // Walk through the directory and tell Cargo to re-run if any .desc files change
+        if let Ok(entries) = fs::read_dir(examples_dir) {
+            for entry in entries.flatten() {
+                if let Some(path) = entry.path().to_str() {
+                    if path.ends_with(".desc") {
+                        println!("cargo:rerun-if-changed={}", path);
+                    }
+                }
+            }
+        }
+    }
+    
+    // Also watch the entire directory for new files
+    println!("cargo:rerun-if-changed={}", examples_dir);
+}

descend_derive/src/lib.rs

@@ -152,8 +152,8 @@ pub fn span_derive(attr: TokenStream, input: TokenStream) -> TokenStream {
 /// Proc macro to automatically generate tests for all .desc files in examples/core/
 #[proc_macro]
 pub fn generate_desc_tests(_input: TokenStream) -> TokenStream {
-    // Use include_dir! to properly track filesystem dependencies
-    // This ensures Cargo knows to re-run the macro when files change
+    // Use include_dir! as a fallback for compile-time file inclusion
+    // This ensures the files are available even if the directory structure changes
     let dir = include_dir::include_dir!("$CARGO_MANIFEST_DIR/../examples/core");
 
     let mut test_functions = Vec::new();
@@ -197,7 +197,7 @@ pub fn generate_desc_tests(_input: TokenStream) -> TokenStream {
                     "use" => "use_stmt".to_string(),
                     "where" => "where_clause".to_string(),
                     "while" => "while_loop".to_string(),
-                    "unit" => "simple_unit".to_string(),
+                    "unit" => "core_unit".to_string(),
                     "func" => "func_call".to_string(),
                     "test_no_main" => "no_main".to_string(),
                     _ => test_name,

examples/core/lazy_load_ub.desc

@@ -0,0 +1,6 @@
+fn memory_load<n: nat, r: prv>(
+    a: &r shrd gpu.global [i16; 16],
+    b: &r uniq gpu.global [i16; 16]
+) -[grid: gpu.grid<X<1>, X<16>>]-> () {
+    ()
+}

examples/core/load_ub.desc

@@ -0,0 +1,6 @@
+fn memory_load<n: nat, r: prv>(
+    a: &r shrd gpu.global [i16; 16]
+) -[grid: gpu.grid<X<1>, X<16>>]-> () {
+    a;
+    ()
+}

examples/core/load_ub_twice.desc

@@ -0,0 +1,8 @@
+fn memory_load<n: nat, r: prv>(
+    a: &r shrd gpu.global [i16; 16],
+    b: &r uniq gpu.global [i16; 16]
+) -[grid: gpu.grid<X<1>, X<16>>]-> () {
+    a;
+    b;
+    ()
+}

examples/core/vec_add.desc.off

@@ -0,0 +1,8 @@
+fn add<n: nat, r: prv>(
+    a: &r shrd gpu.global [i16; 16],
+    b: &r shrd gpu.global [i16; 16],
+    c: &r uniq gpu.global [i16; 16]
+) -[grid: gpu.grid<X<1>, X<16>>]-> () {
+    a + b;
+    ()
+}

examples/core/vec_add_1.desc.off

@@ -0,0 +1,8 @@
+fn add<n: nat, r: prv>(
+    a: &r shrd gpu.global [i16; 16],
+    b: &r shrd gpu.global [i16; 16],
+    c: &r uniq gpu.global [i16; 16]
+) -[grid: gpu.grid<X<1>, X<16>>]-> () {
+    c = a + b;
+    ()
+}

examples/core/vec_add_2.desc.off

@@ -0,0 +1,11 @@
+fn add<n: nat, r: prv>(
+    a: &r shrd gpu.global [i16; 16],
+    b: &r shrd gpu.global [i16; 16],
+    c: &r uniq gpu.global [i16; 16],
+    d: &r uniq gpu.global [i16; 16],
+    e: &r uniq gpu.global [i16; 16]
+) -[grid: gpu.grid<X<1>, X<16>>]-> () {
+    d = a + b;
+    e = b + c;
+    ()
+}

examples/core/vec_add_3.desc.off

@@ -0,0 +1,7 @@
+fn add<n: nat, r: prv>(
+    a: &r shrd gpu.global [i16; 16],
+    b: &r shrd gpu.global [i16; 16]
+) -[grid: gpu.grid<X<1>, X<16>>]-> () {
+    a + b;
+    ()
+}