[Variants] add transopt rules

Author: bongjunj

cranelift/codegen/meta/src/isle.rs

@@ -116,6 +116,7 @@ pub fn get_isle_compilations(
                     src_opts.join("spaceship.isle"),
                     src_opts.join("spectre.isle"),
                     src_opts.join("vector.isle"),
+                    src_opts.join("transopt.isle"),
                 ],
                 untracked_inputs: vec![numerics_isle.clone(), clif_opt_isle],
             },

cranelift/codegen/src/opts/arithmetic.isle

@@ -0,0 +1,35 @@
+;; shootout-random fix
+;; There are arithmetic rules that cannot be transplanted by transopt
+
+;; General cases for `udiv` with constant divisors.
+(rule (simplify_skeleton (udiv x (iconst_u $I32 (u64_extract_non_zero (u32_from_u64 d)))))
+      (if-let false (u32_is_power_of_two d))
+      (apply_div_const_magic_u32 (Opcode.Udiv) x d))
+(rule (simplify_skeleton (udiv x (iconst_u $I64 (u64_extract_non_zero d))))
+      (if-let false (u64_is_power_of_two d))
+      (apply_div_const_magic_u64 (Opcode.Udiv) x d))
+
+;; General cases for `sdiv` with constant divisors.
+(rule (simplify_skeleton (sdiv x (iconst_s $I32 (i64_extract_non_zero (i32_from_i64 d)))))
+      (if-let false (i64_is_any_sign_power_of_two d))
+      (apply_div_const_magic_s32 (Opcode.Sdiv) x d))
+(rule (simplify_skeleton (sdiv x (iconst_s $I64 (i64_extract_non_zero d))))
+      (if-let false (i64_is_any_sign_power_of_two d))
+      (apply_div_const_magic_s64 (Opcode.Sdiv) x d))
+
+;; General cases for `urem` with constant divisors.
+(rule (simplify_skeleton (urem x (iconst_u $I32 (u64_extract_non_zero (u32_from_u64 d)))))
+      (if-let false (u32_is_power_of_two d))
+      (apply_div_const_magic_u32 (Opcode.Urem) x d))
+(rule (simplify_skeleton (urem x (iconst_u $I64 (u64_extract_non_zero d))))
+      (if-let false (u64_is_power_of_two d))
+      (apply_div_const_magic_u64 (Opcode.Urem) x d))
+
+;; General cases for `srem` with constant divisors.
+(rule (simplify_skeleton (srem x (iconst_s $I32 (i64_extract_non_zero (i32_from_i64 d)))))
+      (if-let false (i64_is_any_sign_power_of_two d))
+      (apply_div_const_magic_s32 (Opcode.Srem) x d))
+(rule (simplify_skeleton (srem x (iconst_s $I64 (i64_extract_non_zero d))))
+      (if-let false (i64_is_any_sign_power_of_two d))
+      (apply_div_const_magic_s64 (Opcode.Srem) x d))
+