Brodey/graph contd (#98)

Author: brodeynewman

.github/workflows/format.yml

@@ -3,7 +3,6 @@ on:
   - pull_request
 jobs:
   formatting-check:
-    if: false
     name: Formatting Check
     runs-on: ubuntu-latest
     steps:
@@ -15,22 +14,20 @@ jobs:
           git fetch origin ${{ github.base_ref }}:refs/remotes/origin/${{ github.base_ref }}
       - name: Run clang-format on diff
         run: >
-          # Get the diff of changed files
-
           git diff --name-only origin/${{ github.base_ref }} | grep -E '\.(cpp|h|cu)$' > changed_files.txt || true
 
-          # Check each changed file
-
           while read file; do
             echo "Checking $file"
+            
+            # Get diff in unified format
             git diff origin/${{ github.base_ref }} -- $file | clang-format-diff -p1 -style=file > formatted.diff
+            
             if [ -s formatted.diff ]; then
-              echo "File $file is not properly formatted. Please run clang-format on your code."
-              cat formatted.diff              
-              cat formatted.diff >> "$GITHUB_OUTPUT"
+              echo "File $file is not properly formatted. Please run format on your code."
+              cat formatted.diff
               exit 1
             fi
-          done < changed_files.txt
+          done
       - name: Show success message
         if: success()
         run: echo "All files are properly formatted!"

codegen/annotations.h

@@ -6284,6 +6284,7 @@ cudaError_t cudaGraphExternalSemaphoresWaitNodeSetParams(
     cudaGraphNode_t hNode,
     const struct cudaExternalSemaphoreWaitNodeParams *nodeParams);
 /**
+ * @disabled
  * @param numDependencies SEND_ONLY
  * @param pGraphNode SEND_RECV
  * @param graph SEND_ONLY
@@ -6303,6 +6304,7 @@ cudaError_t
 cudaGraphMemAllocNodeGetParams(cudaGraphNode_t node,
                                struct cudaMemAllocNodeParams *params_out);
 /**
+ * @disabled
  * @param numDependencies SEND_ONLY
  * @param pGraphNode SEND_RECV
  * @param graph SEND_ONLY
@@ -6322,10 +6324,12 @@ cudaError_t cudaGraphMemFreeNodeGetParams(cudaGraphNode_t node, void *dptr_out);
  * @param device SEND_ONLY
  */
 cudaError_t cudaDeviceGraphMemTrim(int device);
+
 /**
+ * @disabled
  * @param device SEND_ONLY
  * @param attr SEND_ONLY
- * @param value SEND_RECV
+ * @param value RECV_ONLY
  */
 cudaError_t cudaDeviceGetGraphMemAttribute(int device,
                                            enum cudaGraphMemAttributeType attr,

codegen/codegen.py

@@ -80,7 +80,10 @@
     "cudaGraphDestroy",
     "cudaGraphAddKernelNode",
     "cudaGraphAddMemcpyNode",
-    "cudaGraphAddHostNode"
+    "cudaGraphAddHostNode",
+    "cudaGraphAddMemFreeNode",
+    "cudaGraphAddMemAllocNode",
+    "cudaDeviceGetGraphMemAttribute"
 ]