fix(cust): handle CUDA 13.3 reserved graph node type

Author: brandonros

crates/cust/build.rs

@@ -10,6 +10,11 @@ fn main() {
     if driver_version >= 12030 {
         println!("cargo::rustc-cfg=conditional_node");
     }
+    // CUDA 13.3 headers add a reserved graph node discriminant.
+    println!("cargo::rustc-check-cfg=cfg(reserved_graph_node_16)");
+    if driver_version >= 13030 {
+        println!("cargo::rustc-cfg=reserved_graph_node_16");
+    }
     // In CUDA 13.0 several pairs/trios of functions were merged:
     // ```
     // CUresult cuMemAdvise(CUdeviceptr devPtr, size_t count, CUmem_advise advice, CUdevice device);

crates/cust/src/graph.rs

@@ -175,6 +175,9 @@ pub enum GraphNodeType {
     /// Conditional node.
     #[cfg(conditional_node)]
     Conditional,
+    /// Reserved by CUDA; this does not represent a supported graph operation.
+    #[cfg(reserved_graph_node_16)]
+    Reserved16,
 }
 
 impl GraphNodeType {
@@ -197,6 +200,8 @@ impl GraphNodeType {
             CU_GRAPH_NODE_TYPE_BATCH_MEM_OP => GraphNodeType::BatchMemoryOperation,
             #[cfg(conditional_node)]
             CU_GRAPH_NODE_TYPE_CONDITIONAL => GraphNodeType::Conditional,
+            #[cfg(reserved_graph_node_16)]
+            CU_GRAPH_NODE_TYPE_RESERVED_16 => GraphNodeType::Reserved16,
         }
     }
 
@@ -219,6 +224,8 @@ impl GraphNodeType {
             Self::BatchMemoryOperation => CU_GRAPH_NODE_TYPE_BATCH_MEM_OP,
             #[cfg(conditional_node)]
             Self::Conditional => CU_GRAPH_NODE_TYPE_CONDITIONAL,
+            #[cfg(reserved_graph_node_16)]
+            Self::Reserved16 => CU_GRAPH_NODE_TYPE_RESERVED_16,
         }
     }
 }
@@ -518,3 +525,26 @@ impl Drop for Graph {
         }
     }
 }
+
+#[cfg(test)]
+mod node_type_tests {
+    use super::{GraphNodeType, driver_sys::CUgraphNodeType};
+
+    #[test]
+    fn kernel_node_round_trip() {
+        let raw = CUgraphNodeType::CU_GRAPH_NODE_TYPE_KERNEL;
+        assert_eq!(
+            GraphNodeType::from_raw(raw),
+            GraphNodeType::KernelInvocation
+        );
+        assert_eq!(GraphNodeType::from_raw(raw).to_raw(), raw);
+    }
+
+    #[cfg(reserved_graph_node_16)]
+    #[test]
+    fn reserved_node_round_trip() {
+        let raw = CUgraphNodeType::CU_GRAPH_NODE_TYPE_RESERVED_16;
+        assert_eq!(GraphNodeType::from_raw(raw), GraphNodeType::Reserved16);
+        assert_eq!(GraphNodeType::from_raw(raw).to_raw(), raw);
+    }
+}