@@ -339,16 +339,8 @@ function simulate_internal(
339
339
1 ,
340
340
)
341
341
342
- weight = MLIR. IR. result (
343
- MLIR. Dialects. builtin. unrealized_conversion_cast (
344
- [MLIR. IR. result (simulate_op, 2 )];
345
- outputs= [MLIR. IR. TensorType (Int64[], MLIR. IR. Type (Float64))],
346
- ),
347
- 1 ,
348
- )
349
-
350
342
trace = TracedRArray {UInt64,0} ((), trace, ())
351
- weight = TracedRArray {Float64,0} ((), weight , ())
343
+ weight = TracedRArray {Float64,0} ((), MLIR . IR . result (simulate_op, 2 ) , ())
352
344
353
345
return trace, weight, result
354
346
end
@@ -507,16 +499,8 @@ function generate_internal(
507
499
1 ,
508
500
)
509
501
510
- weight = MLIR. IR. result (
511
- MLIR. Dialects. builtin. unrealized_conversion_cast (
512
- [MLIR. IR. result (generate_op, 2 )];
513
- outputs= [MLIR. IR. TensorType (Int64[], MLIR. IR. Type (Float64))],
514
- ),
515
- 1 ,
516
- )
517
-
518
502
trace = TracedRArray {UInt64,0} ((), trace, ())
519
- weight = TracedRArray {Float64,0} ((), weight , ())
503
+ weight = TracedRArray {Float64,0} ((), MLIR . IR . result (generate_op, 2 ) , ())
520
504
521
505
return trace, weight, result
522
506
end
0 commit comments