Fix type of preserved_args

mofeing · mofeing · commit 3ecafefd3a2b · 2024-10-05T21:49:14.000-04:00
diff --git a/src/Compiler.jl b/src/Compiler.jl
@@ -288,7 +288,7 @@ function compile_mlir!(mod, f, args; optimize=true)
         )
     end
 
-    preserved_args = Tuple{TracedRArray,Int}[]
+    preserved_args = Tuple{TracedTypes,Int}[]
     results = [MLIR.IR.operand(ret, i) for i in 1:MLIR.IR.noperands(ret)]
     nresults = MLIR.IR.Value[]
     linear_results2 = TracedTypes[]

Original file line number	Diff line number	Diff line change
`@@ -288,7 +288,7 @@ function compile_mlir!(mod, f, args; optimize=true)`
`288`	`288`	`)`
`289`	`289`	`end`
`290`	`290`
`291`		`- preserved_args = Tuple{TracedRArray,Int}[]`
	`291`	`+ preserved_args = Tuple{TracedTypes,Int}[]`
`292`	`292`	`results = [MLIR.IR.operand(ret, i) for i in 1:MLIR.IR.noperands(ret)]`
`293`	`293`	`nresults = MLIR.IR.Value[]`
`294`	`294`	`linear_results2 = TracedTypes[]`