From 3322ce3d08f9619843286b86bb57c4c515d01b2b Mon Sep 17 00:00:00 2001
From: root <melchert@stanford.edu>
Date: Sun, 26 May 2024 12:38:36 -0700
Subject: [PATCH] Fixed scomp issues

---
 src/CodeGen_Pono.cpp | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/CodeGen_Pono.cpp b/src/CodeGen_Pono.cpp
index fbf49729f..5a26ee250 100644
--- a/src/CodeGen_Pono.cpp
+++ b/src/CodeGen_Pono.cpp
@@ -960,7 +960,8 @@ string associated_provide_name(Stmt s, string call_name) {
                 stream << ')';
             } else {
                 stream << "solver.create_term(solver.ops.Ite, ";
-                    if (op->type.is_int()) {
+                Halide::Type t = op->a.type();
+                    if (t.is_int()) {
                         stream << "solver.create_term(solver.ops.BVSlt, ";
                     } else {
                         stream << "solver.create_term(solver.ops.BVUlt, ";
@@ -982,7 +983,8 @@ string associated_provide_name(Stmt s, string call_name) {
                 stream << ')';
             } else {
                 stream << "solver.create_term(solver.ops.Ite, ";
-                    if (op->type.is_int()) {
+                Halide::Type t = op->a.type();
+                    if (t.is_int()) {
                         stream << "solver.create_term(solver.ops.BVSle, ";
                     } else {
                         stream << "solver.create_term(solver.ops.BVUle, ";
@@ -1004,7 +1006,8 @@ string associated_provide_name(Stmt s, string call_name) {
                 stream << ')';
             } else {
                 stream << "solver.create_term(solver.ops.Ite, ";
-                    if (op->type.is_int()) {
+                Halide::Type t = op->a.type();
+                    if (t.is_int()) {
                         stream << "solver.create_term(solver.ops.BVSgt, ";
                     } else {
                         stream << "solver.create_term(solver.ops.BVUgt, ";
@@ -1026,7 +1029,8 @@ string associated_provide_name(Stmt s, string call_name) {
                 stream << ')';
             } else {
                 stream << "solver.create_term(solver.ops.Ite, ";
-                    if (op->type.is_int()) {
+                Halide::Type t = op->a.type();
+                    if (t.is_int()) {
                         stream << "solver.create_term(solver.ops.BVSge, ";
                     } else {
                         stream << "solver.create_term(solver.ops.BVUge, ";