fix for new type guard

Author: bongjunj

cranelift/codegen/src/isle_prelude.rs

@@ -533,6 +533,15 @@ macro_rules! isle_common_prelude_methods {
             ty.is_int().then(|| ty)
         }
 
+        #[inline]
+        fn ty_int_vec128(&mut self, ty: Type) -> Option<Type> {
+            if ty.is_int() || (ty.is_vector() && ty.bits() == 128 && ty.lane_type().is_int()) {
+                Some(ty)
+            } else {
+                None
+            }
+        }
+
         #[inline]
         fn ty_scalar(&mut self, ty: Type) -> Option<Type> {
             if ty.lane_count() == 1 { Some(ty) } else { None }

cranelift/codegen/src/prelude.isle

@@ -505,6 +505,9 @@
 (decl ty_vec128 (Type) Type)
 (extern extractor ty_vec128 ty_vec128)
 
+(decl ty_int_vec128 (Type) Type)
+(extern extractor ty_int_vec128 ty_int_vec128)
+
 ;; An extractor that only matches dynamic vector types with a 64-bit
 ;; base type.
 (decl ty_dyn_vec64 (Type) Type)

cranelift/codegen/src/prelude_opt.isle

@@ -142,6 +142,7 @@
     (if-let true (i64_eq c c_reextended))
     (iconst ty (imm64 c_masked)))
 (rule 1 (iconst_s $I128 c) (sextend $I128 (iconst_s $I64 c)))
+(rule 2 (iconst_s (ty_vec128 ty) c) (splat ty (iconst_s (lane_type ty) c)))
 
 ;; Construct an `iconst` from a `u64` or Extract a `u64` from an `iconst`
 ;; by treating the constant as unsigned.
@@ -157,6 +158,7 @@
     (if-let true (u64_lt_eq c (ty_umax ty)))
     (iconst ty (imm64 c)))
 (rule 1 (iconst_u $I128 c) (uextend $I128 (iconst_u $I64 c)))
+(rule 2 (iconst_u (ty_vec128 ty) c) (splat ty (iconst_u (lane_type ty) c)))
 
 ;; These take `Value`, rather than going through `inst_data_value_tupled`,
 ;; because most of the time they want to return the original `Value`, and it