fix(cust): add cuMemLocation_anon_id cfg guard for CUDA 13.0 compat

Made-with: Cursor

Author: CharryWu

crates/cust/build.rs

@@ -40,4 +40,12 @@ fn main() {
         println!("cargo::rustc-cfg=cuGraphGetEdges_v2");
         println!("cargo::rustc-cfg=cuCtxCreate_v4");
     }
+
+    // In CUDA 13.2 the `id` field in `CUmemLocation_st` was placed inside an anonymous union.
+    // Bindgen renders this as `__bindgen_anon_1: CUmemLocation_st__bindgen_ty_1` instead of a
+    // direct `id` field. This cfg gates the struct initialization syntax accordingly.
+    println!("cargo::rustc-check-cfg=cfg(cuMemLocation_anon_id)");
+    if driver_version >= 13020 {
+        println!("cargo::rustc-cfg=cuMemLocation_anon_id");
+    }
 }

crates/cust/src/memory/unified.rs

@@ -647,7 +647,10 @@ pub trait MemoryAdvise<T: DeviceCopy>: private::Sealed {
                 #[cfg(cuMemPrefetchAsync_v2)]
                 driver_sys::CUmemLocation {
                     type_: driver_sys::CUmemLocationType::CU_MEM_LOCATION_TYPE_DEVICE,
+                    #[cfg(cuMemLocation_anon_id)]
                     __bindgen_anon_1: driver_sys::CUmemLocation_st__bindgen_ty_1 { id },
+                    #[cfg(not(cuMemLocation_anon_id))]
+                    id,
                 },
                 #[cfg(not(cuMemPrefetchAsync_v2))]
                 id,
@@ -693,7 +696,10 @@ pub trait MemoryAdvise<T: DeviceCopy>: private::Sealed {
                 #[cfg(cuMemPrefetchAsync_v2)]
                 driver_sys::CUmemLocation {
                     type_: driver_sys::CUmemLocationType::CU_MEM_LOCATION_TYPE_DEVICE,
+                    #[cfg(cuMemLocation_anon_id)]
                     __bindgen_anon_1: driver_sys::CUmemLocation_st__bindgen_ty_1 { id },
+                    #[cfg(not(cuMemLocation_anon_id))]
+                    id,
                 },
                 #[cfg(not(cuMemPrefetchAsync_v2))]
                 id,
@@ -735,7 +741,10 @@ pub trait MemoryAdvise<T: DeviceCopy>: private::Sealed {
                 #[cfg(cuMemAdvise_v2)]
                 driver_sys::CUmemLocation {
                     type_: driver_sys::CUmemLocationType::CU_MEM_LOCATION_TYPE_DEVICE,
+                    #[cfg(cuMemLocation_anon_id)]
                     __bindgen_anon_1: driver_sys::CUmemLocation_st__bindgen_ty_1 { id },
+                    #[cfg(not(cuMemLocation_anon_id))]
+                    id,
                 },
                 #[cfg(not(cuMemAdvise_v2))]
                 id,
@@ -777,7 +786,10 @@ pub trait MemoryAdvise<T: DeviceCopy>: private::Sealed {
                 #[cfg(cuMemAdvise_v2)]
                 driver_sys::CUmemLocation {
                     type_: driver_sys::CUmemLocationType::CU_MEM_LOCATION_TYPE_DEVICE,
+                    #[cfg(cuMemLocation_anon_id)]
                     __bindgen_anon_1: driver_sys::CUmemLocation_st__bindgen_ty_1 { id },
+                    #[cfg(not(cuMemLocation_anon_id))]
+                    id,
                 },
                 #[cfg(not(cuMemAdvise_v2))]
                 id,
@@ -801,7 +813,10 @@ pub trait MemoryAdvise<T: DeviceCopy>: private::Sealed {
                 #[cfg(cuMemAdvise_v2)]
                 driver_sys::CUmemLocation {
                     type_: driver_sys::CUmemLocationType::CU_MEM_LOCATION_TYPE_DEVICE,
+                    #[cfg(cuMemLocation_anon_id)]
                     __bindgen_anon_1: driver_sys::CUmemLocation_st__bindgen_ty_1 { id },
+                    #[cfg(not(cuMemLocation_anon_id))]
+                    id,
                 },
                 #[cfg(not(cuMemAdvise_v2))]
                 id,