Update TypeTree tests to verify metadata attachment

  - Fix nott-flag test to emit LLVM IR and check enzyme_type attributes
  - Replace TODO comments with actual TypeTree metadata verification
  - Test that NoTT flag properly disables TypeTree generation
  - Test that TypeTree enabled generates proper enzyme_type attributes

Signed-off-by: Karan Janthe <[email protected]>

Author: KMJ-007

tests/run-make/autodiff/type-trees/nott-flag/nott.check

@@ -1,3 +1,5 @@
-// TODO(KMJ-007): Update this test when TypeTree integration is complete
-// CHECK: square - {[-1]:Float@double} |{[-1]:Pointer}:{}
-// CHECK: ptr %{{[0-9]+}}: {[-1]:Pointer, [-1,0]:Float@double}
+// Check that enzyme_type attributes are NOT present when NoTT flag is used
+// This verifies the NoTT flag correctly disables TypeTree metadata
+
+CHECK: define{{.*}}@square
+CHECK-NOT: "enzyme_type"

tests/run-make/autodiff/type-trees/nott-flag/rmake.rs

@@ -5,34 +5,32 @@ use run_make_support::{llvm_filecheck, rfs, rustc};
 
 fn main() {
     // Test with NoTT flag - should not generate TypeTree metadata
-    let output_nott = rustc()
+    rustc()
         .input("test.rs")
-        .arg("-Zautodiff=Enable,NoTT,PrintTAFn=square")
-        .arg("-Zautodiff=NoPostopt")
-        .opt_level("3")
-        .arg("-Clto=fat")
-        .arg("-g")
+        .arg("-Zautodiff=Enable,NoTT")
+        .emit("llvm-ir")
+        .arg("-o")
+        .arg("nott.ll")
         .run();
 
-    // Write output for NoTT case
-    rfs::write("nott.stdout", output_nott.stdout_utf8());
-    
     // Test without NoTT flag - should generate TypeTree metadata
-    let output_with_tt = rustc()
+    rustc()
         .input("test.rs")
-        .arg("-Zautodiff=Enable,PrintTAFn=square")
-        .arg("-Zautodiff=NoPostopt")
-        .opt_level("3")
-        .arg("-Clto=fat")
-        .arg("-g")
+        .arg("-Zautodiff=Enable")
+        .emit("llvm-ir")
+        .arg("-o")
+        .arg("with_tt.ll")
         .run();
 
-    // Write output for TypeTree case  
-    rfs::write("with_tt.stdout", output_with_tt.stdout_utf8());
-
-    // Verify NoTT output has minimal TypeTree info
-    llvm_filecheck().patterns("nott.check").stdin_buf(rfs::read("nott.stdout")).run();
+    // Verify NoTT version does NOT have enzyme_type attributes
+    llvm_filecheck()
+        .patterns("nott.check")
+        .stdin_buf(rfs::read("nott.ll"))
+        .run();
 
-    // Verify normal output will have TypeTree info (once implemented)
-    llvm_filecheck().patterns("with_tt.check").stdin_buf(rfs::read("with_tt.stdout")).run();
+    // Verify TypeTree version DOES have enzyme_type attributes
+    llvm_filecheck()
+        .patterns("with_tt.check")
+        .stdin_buf(rfs::read("with_tt.ll"))
+        .run();
 }

tests/run-make/autodiff/type-trees/nott-flag/with_tt.check

@@ -1,3 +1,4 @@
-// TODO(KMJ-007): Update this test when TypeTree integration is complete
-// CHECK: square - {[-1]:Float@double} |{[-1]:Pointer}:{}
-// CHECK: ptr %{{[0-9]+}}: {[-1]:Pointer, [-1,0]:Float@double}
+// Check that enzyme_type attributes are present when TypeTree is enabled
+// This verifies our TypeTree metadata attachment is working
+
+CHECK: define{{.*}}"enzyme_type"="{[]:Float@double}"{{.*}}@square{{.*}}"enzyme_type"="{[]:Pointer}"