← All problems Problem 632
P(N) = product of non-zero C_k(N) mod 10^9+7, N=10^16. C_k(N) = count of integers 1..N with exactly k square prime factors. Uses linear sieve for mu/omega, recursive squarefree counting, binomial inversion.
View problem on Project Euler
Performance comparison
Metric Our solution Best known
Time complexity O(n^2)O(sqrt(n))Space complexity O(n^2)O(1)Approach Flow solution Trial division or Pollard rho Verdict Suboptimal
Flow source
# Project Euler 632: Square Prime Factors
# P(N) = product of non-zero C_k(N) mod 10^9+7, N=10^16.
# C_k(N) = count of integers 1..N with exactly k square prime factors.
# Uses linear sieve for mu/omega, recursive squarefree counting, binomial inversion.
import euler.nt { isqrt }
extern {
function calloc(n: i64, size: i64) -> ptr<void>
function free(p: ptr<void>) -> void
function memset(s: ptr<void>, c: i64, n: i64) -> ptr<void>
}
const MOD: i64 = 1000000007
# Globals for recursive squarefree counting
let mut g_primes: ptr<i64> = null
let mut g_nprimes: i64 = 0
function pi(x: i64) -> i64 {
# Binary search for number of primes <= x
let mut lo: i64 = 0
let mut hi: i64 = g_nprimes
while lo < hi {
let mid: i64 = (lo + hi) / 2
if g_primes[mid] <= x { lo = mid + 1 } else { hi = mid }
}
return lo
}
function kth_root_floor(n: i64, k: i64) -> i64 {
if k == 2 { return isqrt(n) }
# Binary search for k-th root
let mut lo: i64 = 1
let mut hi: i64 = 2
# Find upper bound
while hi > 0 {
let mut p: i64 = 1
let mut ok: i64 = 1
for _ in 0..k {
if p > n / hi { ok = 0 }
if ok == 1 { p = p * hi }
}
if ok == 0 { break }
hi = hi * 2
if hi > 1000000000 { break }
}
while lo + 1 < hi {
let mid: i64 = (lo + hi) / 2
let mut p: i64 = 1
let mut ok: i64 = 1
for _ in 0..k {
if p > n / mid { ok = 0 }
if ok == 1 { p = p * mid }
}
if ok == 1 { lo = mid } else { hi = mid }
}
return lo
}
# Memoized count_squarefree_products(x, k, start)
# Count of numbers <= x that are products of exactly k distinct primes
# from primes[start:], strictly increasing.
const MEMO_SIZE: i64 = 4000007
let mut g_memo_key: ptr<i64> = null
let mut g_memo_val: ptr<i64> = null
function memo_key(x: i64, k: i64, start: i64) -> i64 {
return x * 1000 + k * 100 + start
}
function count_sqfree(x: i64, k: i64, start: i64) -> i64 {
if k == 0 { return 1 }
if k == 1 { return pi(x) - start }
let limit: i64 = kth_root_floor(x, k)
# Binary search for end = number of primes <= limit
let end: i64 = pi(limit)
if end <= start { return 0 }
# Check memo
let key: i64 = memo_key(x, k, start)
let h: i64 = key % MEMO_SIZE
if g_memo_key[h] == key { return g_memo_val[h] }
let mut res: i64 = 0
for i in start..end {
let p: i64 = g_primes[i]
res = res + count_sqfree(x / p, k - 1, i + 1)
}
g_memo_key[h] = key
g_memo_val[h] = res
return res
}
function prefix_count(x: i64, r: i64) -> i64 {
if r == 0 {
if x >= 1 { return 1 }
return 0
}
return count_sqfree(x, r, 0)
}
function main() -> i32 {
let N: i64 = 10000000000000000 # 10^16
let sqrtN: i64 = isqrt(N)
# Sieve primes up to sqrtN (10^8)
# Use odd-only byte sieve
let sieve_size: i64 = (sqrtN + 1) / 2
let sieve: ptr<i8> = calloc(sieve_size, 1)
for i in 0..sieve_size { sieve[i] = 1 }
sieve[0] = 0 # 1 is not prime
let limit: i64 = isqrt(sqrtN)
for i in 1..(limit / 2 + 1) {
if sieve[i] == 1 {
let p: i64 = 2 * i + 1
let mut j: i64 = (p * p) / 2
while j < sieve_size {
sieve[j] = 0
j = j + p
}
}
}
# Count primes and store
let mut nprimes: i64 = 1 # prime 2
for i in 1..sieve_size {
if sieve[i] == 1 { nprimes = nprimes + 1 }
}
g_primes = calloc(nprimes, 8)
g_primes[0] = 2
let mut pidx: i64 = 1
for i in 1..sieve_size {
if sieve[i] == 1 {
g_primes[pidx] = 2 * i + 1
pidx = pidx + 1
}
}
g_nprimes = nprimes
free(sieve)
# Init memo for count_sqfree
g_memo_key = calloc(MEMO_SIZE, 8)
g_memo_val = calloc(MEMO_SIZE, 8)
# Small primes for recursion (up to 10000)
let mut nsmall: i64 = 0
for i in 0..nprimes {
if g_primes[i] <= 10000 { nsmall = nsmall + 1 }
else { break }
}
# max_possible_k: product of first k primes <= sqrtN
let mut prod: i64 = 1
let mut kmax: i64 = 0
for i in 0..nsmall {
if prod <= sqrtN / g_primes[i] {
prod = prod * g_primes[i]
kmax = kmax + 1
} else {
break
}
}
# Linear sieve for mu and omega up to L0
let L0: i64 = 5000000
if L0 > sqrtN { kmax = kmax } # L0 cap
let mu: ptr<i8> = calloc(L0 + 1, 1)
let omega: ptr<i8> = calloc(L0 + 1, 1)
let is_comp: ptr<i8> = calloc(L0 + 1, 1)
let lprimes: ptr<i64> = calloc(L0 / 10 + 100, 8)
let mut nlprimes: i64 = 0
mu[1] = 1
for i in 2..(L0 + 1) {
if is_comp[i] == 0 {
lprimes[nlprimes] = i
nlprimes = nlprimes + 1
mu[i] = -1
omega[i] = 1
}
for j in 0..nlprimes {
let p: i64 = lprimes[j]
let ip: i64 = i * p
if ip > L0 { break }
is_comp[ip] = 1
if i % p == 0 {
mu[ip] = 0
omega[ip] = omega[i]
break
} else {
mu[ip] = -mu[i]
omega[ip] = omega[i] + 1
}
}
}
free(is_comp)
free(lprimes)
# Compute S[r] = sum_{squarefree d, omega(d)=r} floor(N / d^2)
let S: ptr<i64> = calloc(kmax + 1, 8)
# Small d up to L0
for d in 1..(L0 + 1) {
if mu[d] != 0 {
let r: i64 = omega[d] as i64
if r <= kmax {
S[r] = S[r] + N / ((d as i64) * (d as i64))
}
}
}
free(mu)
free(omega)
# For d > L0, q = floor(N/d^2) is small
let qmax: i64 = N / ((L0 + 1) * (L0 + 1))
for q in 1..(qmax + 1) {
let hi: i64 = isqrt(N / q)
let lo: i64 = isqrt(N / (q + 1))
let a: i64 = lo
if L0 > a { a = L0 }
if hi <= a { continue }
for r in 0..(kmax + 1) {
let cnt: i64 = prefix_count(hi, r) - prefix_count(a, r)
if cnt != 0 {
S[r] = S[r] + q * cnt
}
}
}
# Binomial inversion: C[k] = S[k] - sum_{j>k} C[j] * comb[j][k]
let comb: ptr<i64> = calloc((kmax + 1) * (kmax + 1), 8)
for n in 0..(kmax + 1) {
comb[n * (kmax + 1) + 0] = 1
comb[n * (kmax + 1) + n] = 1
for r in 1..n {
comb[n * (kmax + 1) + r] = comb[(n - 1) * (kmax + 1) + r - 1] + comb[(n - 1) * (kmax + 1) + r]
}
}
let C: ptr<i64> = calloc(kmax + 1, 8)
for k in 0..(kmax + 1) {
let rev_k: i64 = kmax - k
let mut val: i64 = S[rev_k]
for j in (rev_k + 1)..(kmax + 1) {
val = val - C[j] * comb[j * (kmax + 1) + rev_k]
}
C[rev_k] = val
}
# Product of non-zero C[k] mod MOD
let mut ans: i64 = 1
for k in 0..(kmax + 1) {
if C[k] != 0 {
let v: i64 = C[k] % MOD
if v < 0 { v = v + MOD }
ans = (ans * v) % MOD
}
}
printf("%lld\n", ans)
free(C)
free(comb)
free(S)
free(g_memo_val)
free(g_memo_key)
free(g_primes)
return 0
}
Generated C
#include <stdint.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* Flow runtime helpers */
typedef struct flow_temp_node { struct flow_temp_node* next; } flow_temp_node;
static flow_temp_node* flow_temp_head = NULL;
static int flow_temp_atexit_set = 0;
__attribute__((unused)) static void flow_temp_free_all(void) {
while (flow_temp_head) {
flow_temp_node* n = flow_temp_head;
flow_temp_head = n->next;
free(n);
}
}
__attribute__((unused)) static void* flow_temp_alloc(size_t nbytes) {
flow_temp_node* node = (flow_temp_node*)malloc(sizeof(flow_temp_node) + nbytes);
if (!node) return NULL;
node->next = flow_temp_head;
flow_temp_head = node;
if (!flow_temp_atexit_set) {
flow_temp_atexit_set = 1;
atexit(flow_temp_free_all);
}
return (void*)(node + 1);
}
#ifndef FLOW_DIAG
#define FLOW_DIAG(msg) fprintf(stderr, "%s", (msg))
#endif
#ifndef FLOW_LOG
#define FLOW_LOG(fmt, ...) printf(fmt, __VA_ARGS__)
#endif
#ifndef FLOW_LOG_EMPTY
#define FLOW_LOG_EMPTY(fmt) printf(fmt)
#endif
static char* flow_strcat(const char* a, const char* b) {
size_t la = strlen(a ? a : ""), lb = strlen(b ? b : "");
char* r = (char*)flow_temp_alloc(la + lb + 1);
if (!r) return NULL;
if (la) memcpy(r, a, la);
if (lb) memcpy(r + la, b, lb);
r[la + lb] = '\0';
return r;
}
#define __flow_in_arr(arr, val) __extension__ ({ \
int _found = 0; \
size_t _n = sizeof(arr)/sizeof((arr)[0]); \
for (size_t _i = 0; _i < _n; _i++) { \
if ((arr)[_i] == (val)) { _found = 1; break; } \
} _found; })
/* Unified fault handler (MISRA #279) — override with -DFLOW_FAULT_HANDLER=fn */
#ifndef FLOW_FAULT_HANDLER
__attribute__((unused)) static inline void flow_fault_handler(const char* msg) {
fprintf(stderr, "flow: %s\n", msg ? msg : "fault");
abort();
#if defined(__GNUC__) || defined(__clang__)
__builtin_unreachable();
#endif
}
#else
#define flow_fault_handler FLOW_FAULT_HANDLER
#endif
#define flow_div_by_zero_handler() flow_fault_handler("division by zero")
#define flow_shift_ub_handler() flow_fault_handler("invalid shift (amount out of range or left-shift of negative)")
#ifndef FLOW_CHECKED_DIV
#define FLOW_CHECKED_DIV(L, R) (((R) != 0) ? ((L) / (R)) : (flow_div_by_zero_handler(), (L) * 0))
#endif
#ifndef FLOW_CHECKED_MOD
#define FLOW_CHECKED_MOD(L, R) (((R) != 0) ? ((L) % (R)) : (flow_div_by_zero_handler(), (L) * 0))
#endif
#ifndef FLOW_CHECKED_SHL
#define FLOW_CHECKED_SHL(L, R) ((((R) >= 0) && ((unsigned long long)(R) < (sizeof(L) * 8ull)) && ((L) >= 0)) ? ((L) << (R)) : (flow_shift_ub_handler(), (L) * 0))
#endif
#ifndef FLOW_CHECKED_SHR
#define FLOW_CHECKED_SHR(L, R) ((((R) >= 0) && ((unsigned long long)(R) < (sizeof(L) * 8ull))) ? ((L) >> (R)) : (flow_shift_ub_handler(), (L) * 0))
#endif
#include <math.h>
void* _ui_state = NULL;
static inline float i32_to_f32(int32_t v) { return (float)v; }
/* Host stub for @gpu kernels (device codegen replaces this). */
static inline int32_t gpu_thread_id(void) { return 0; }
int64_t gcd_i64_i64(int64_t a0, int64_t b0);
int64_t lcm_i64_i64(int64_t a, int64_t b);
int64_t isqrt_i64(int64_t n);
int64_t mulmod_i64_i64_i64(int64_t a0, int64_t b0, int64_t mod);
int64_t mod_pow_i64_i64_i64(int64_t base, int64_t exp, int64_t mod);
bool is_prime_i64(int64_t n);
int64_t pi_i64(int64_t x);
int64_t kth_root_floor_i64_i64(int64_t n, int64_t k);
int64_t memo_key_i64_i64_i64(int64_t x, int64_t k, int64_t start);
int64_t count_sqfree_i64_i64_i64(int64_t x, int64_t k, int64_t start);
int64_t prefix_count_i64_i64(int64_t x, int64_t r);
int32_t main(void);
static const int64_t MOD = 1000000007;
static const int64_t MEMO_SIZE = 4000007;
/* Module statics */
static int64_t* g_primes = NULL;
static int64_t g_nprimes = 0;
static int64_t* g_memo_key = NULL;
static int64_t* g_memo_val = NULL;
int64_t gcd_i64_i64(int64_t a0, int64_t b0) {
int64_t a = a0;
int64_t b = b0;
while (b != 0) {
int64_t t = FLOW_CHECKED_MOD((a), (b));
a = b;
b = t;
}
return a;
}
int64_t lcm_i64_i64(int64_t a, int64_t b) {
if ((a == 0 || b == 0)) {
return 0;
}
return (FLOW_CHECKED_DIV((a), (gcd_i64_i64(a, b))) * b);
}
int64_t isqrt_i64(int64_t n) {
if (n < 2) {
return n;
}
int64_t x = n;
int64_t y = FLOW_CHECKED_DIV(((x + 1)), (2));
while (y < x) {
x = y;
y = FLOW_CHECKED_DIV(((x + FLOW_CHECKED_DIV((n), (x)))), (2));
}
return x;
}
int64_t mulmod_i64_i64_i64(int64_t a0, int64_t b0, int64_t mod) {
int64_t a = FLOW_CHECKED_MOD((a0), (mod));
int64_t b = FLOW_CHECKED_MOD((b0), (mod));
int64_t result = 0;
while (b > 0) {
if (FLOW_CHECKED_MOD((b), (2)) == 1) {
result = FLOW_CHECKED_MOD(((result + a)), (mod));
}
a = FLOW_CHECKED_MOD(((a * 2)), (mod));
b = FLOW_CHECKED_DIV((b), (2));
}
return result;
}
int64_t mod_pow_i64_i64_i64(int64_t base, int64_t exp, int64_t mod) {
if (mod == 1) {
return 0;
}
int64_t result = 1;
int64_t b = FLOW_CHECKED_MOD((base), (mod));
int64_t e = exp;
while (e > 0) {
if (FLOW_CHECKED_MOD((e), (2)) == 1) {
result = mulmod_i64_i64_i64(result, b, mod);
}
b = mulmod_i64_i64_i64(b, b, mod);
e = FLOW_CHECKED_DIV((e), (2));
}
return result;
}
bool is_prime_i64(int64_t n) {
if (n < 2) {
return 0;
}
if (n < 4) {
return 1;
}
if ((FLOW_CHECKED_MOD((n), (2)) == 0 || FLOW_CHECKED_MOD((n), (3)) == 0)) {
return 0;
}
int64_t i = 5;
while ((i * i) <= n) {
if ((FLOW_CHECKED_MOD((n), (i)) == 0 || FLOW_CHECKED_MOD((n), ((i + 2))) == 0)) {
return 0;
}
i = (i + 6);
}
return 1;
}
int64_t pi_i64(int64_t x) {
int64_t lo = 0;
int64_t hi = g_nprimes;
while (lo < hi) {
int64_t mid = FLOW_CHECKED_DIV(((lo + hi)), (2));
if (g_primes[mid] <= x) {
lo = (mid + 1);
} else {
hi = mid;
}
}
return lo;
}
int64_t kth_root_floor_i64_i64(int64_t n, int64_t k) {
if (k == 2) {
return isqrt_i64(n);
}
int64_t lo = 1;
int64_t hi = 2;
while (hi > 0) {
int64_t p = 1;
int64_t ok = 1;
int32_t __flow_step_1 = 1;
for (int32_t _ = 0; (0 <= k) ? _ < k : _ > k; _ += (0 <= k) ? 1 : -1) {
if (p > FLOW_CHECKED_DIV((n), (hi))) {
ok = 0;
}
if (ok == 1) {
p = (p * hi);
}
}
if (ok == 0) {
break;
}
hi = (hi * 2);
if (hi > 1000000000) {
break;
}
}
while ((lo + 1) < hi) {
int64_t mid = FLOW_CHECKED_DIV(((lo + hi)), (2));
int64_t p = 1;
int64_t ok = 1;
int32_t __flow_step_2 = 1;
for (int32_t _ = 0; (0 <= k) ? _ < k : _ > k; _ += (0 <= k) ? 1 : -1) {
if (p > FLOW_CHECKED_DIV((n), (mid))) {
ok = 0;
}
if (ok == 1) {
p = (p * mid);
}
}
if (ok == 1) {
lo = mid;
} else {
hi = mid;
}
}
return lo;
}
int64_t memo_key_i64_i64_i64(int64_t x, int64_t k, int64_t start) {
return (((x * 1000) + (k * 100)) + start);
}
int64_t count_sqfree_i64_i64_i64(int64_t x, int64_t k, int64_t start) {
if (k == 0) {
return 1;
}
if (k == 1) {
return (pi_i64(x) - start);
}
int64_t limit = kth_root_floor_i64_i64(x, k);
int64_t end = pi_i64(limit);
if (end <= start) {
return 0;
}
int64_t key = memo_key_i64_i64_i64(x, k, start);
int64_t h = FLOW_CHECKED_MOD((key), (MEMO_SIZE));
if (g_memo_key[h] == key) {
return g_memo_val[h];
}
int64_t res = 0;
int32_t __flow_step_3 = 1;
for (int32_t i = start; (start <= end) ? i < end : i > end; i += (start <= end) ? 1 : -1) {
int64_t p = g_primes[i];
res = (res + count_sqfree_i64_i64_i64(FLOW_CHECKED_DIV((x), (p)), (k - 1), (i + 1)));
}
g_memo_key[h] = key;
g_memo_val[h] = res;
return res;
}
int64_t prefix_count_i64_i64(int64_t x, int64_t r) {
if (r == 0) {
if (x >= 1) {
return 1;
}
return 0;
}
return count_sqfree_i64_i64_i64(x, r, 0);
}
int32_t main(void) {
int64_t N = 10000000000000000;
int64_t sqrtN = isqrt_i64(N);
int64_t sieve_size = FLOW_CHECKED_DIV(((sqrtN + 1)), (2));
int8_t* sieve = (int8_t*)(calloc(sieve_size, 1));
int32_t __flow_step_4 = 1;
for (int32_t i = 0; (0 <= sieve_size) ? i < sieve_size : i > sieve_size; i += (0 <= sieve_size) ? 1 : -1) {
sieve[i] = 1;
}
sieve[0] = 0;
int64_t limit = isqrt_i64(sqrtN);
int32_t __flow_step_5 = 1;
for (int32_t i = 1; (1 <= (FLOW_CHECKED_DIV((limit), (2)) + 1)) ? i < (FLOW_CHECKED_DIV((limit), (2)) + 1) : i > (FLOW_CHECKED_DIV((limit), (2)) + 1); i += (1 <= (FLOW_CHECKED_DIV((limit), (2)) + 1)) ? 1 : -1) {
if (sieve[i] == 1) {
int64_t p = ((2 * i) + 1);
int64_t j = FLOW_CHECKED_DIV(((p * p)), (2));
while (j < sieve_size) {
sieve[j] = 0;
j = (j + p);
}
}
}
int64_t nprimes = 1;
int32_t __flow_step_6 = 1;
for (int32_t i = 1; (1 <= sieve_size) ? i < sieve_size : i > sieve_size; i += (1 <= sieve_size) ? 1 : -1) {
if (sieve[i] == 1) {
nprimes = (nprimes + 1);
}
}
g_primes = calloc(nprimes, 8);
g_primes[0] = 2;
int64_t pidx = 1;
int32_t __flow_step_7 = 1;
for (int32_t i = 1; (1 <= sieve_size) ? i < sieve_size : i > sieve_size; i += (1 <= sieve_size) ? 1 : -1) {
if (sieve[i] == 1) {
g_primes[pidx] = ((2 * i) + 1);
pidx = (pidx + 1);
}
}
g_nprimes = nprimes;
free(sieve);
g_memo_key = calloc(MEMO_SIZE, 8);
g_memo_val = calloc(MEMO_SIZE, 8);
int64_t nsmall = 0;
int32_t __flow_step_8 = 1;
for (int32_t i = 0; (0 <= nprimes) ? i < nprimes : i > nprimes; i += (0 <= nprimes) ? 1 : -1) {
if (g_primes[i] <= 10000) {
nsmall = (nsmall + 1);
} else {
break;
}
}
int64_t prod = 1;
int64_t kmax = 0;
int32_t __flow_step_9 = 1;
for (int32_t i = 0; (0 <= nsmall) ? i < nsmall : i > nsmall; i += (0 <= nsmall) ? 1 : -1) {
if (prod <= FLOW_CHECKED_DIV((sqrtN), (g_primes[i]))) {
prod = (prod * g_primes[i]);
kmax = (kmax + 1);
} else {
break;
}
}
int64_t L0 = 5000000;
if (L0 > sqrtN) {
kmax = kmax;
}
int8_t* mu = (int8_t*)(calloc((L0 + 1), 1));
int8_t* omega = (int8_t*)(calloc((L0 + 1), 1));
int8_t* is_comp = (int8_t*)(calloc((L0 + 1), 1));
int64_t* lprimes = (int64_t*)(calloc((FLOW_CHECKED_DIV((L0), (10)) + 100), 8));
int64_t nlprimes = 0;
mu[1] = 1;
int32_t __flow_step_10 = 1;
for (int32_t i = 2; (2 <= (L0 + 1)) ? i < (L0 + 1) : i > (L0 + 1); i += (2 <= (L0 + 1)) ? 1 : -1) {
if (is_comp[i] == 0) {
lprimes[nlprimes] = i;
nlprimes = (nlprimes + 1);
mu[i] = (-1);
omega[i] = 1;
}
int32_t __flow_step_11 = 1;
for (int32_t j = 0; (0 <= nlprimes) ? j < nlprimes : j > nlprimes; j += (0 <= nlprimes) ? 1 : -1) {
int64_t p = lprimes[j];
int64_t ip = (i * p);
if (ip > L0) {
break;
}
is_comp[ip] = 1;
if (FLOW_CHECKED_MOD((i), (p)) == 0) {
mu[ip] = 0;
omega[ip] = omega[i];
break;
} else {
mu[ip] = (-mu[i]);
omega[ip] = (omega[i] + 1);
}
}
}
free(is_comp);
free(lprimes);
int64_t* S = (int64_t*)(calloc((kmax + 1), 8));
int32_t __flow_step_12 = 1;
for (int32_t d = 1; (1 <= (L0 + 1)) ? d < (L0 + 1) : d > (L0 + 1); d += (1 <= (L0 + 1)) ? 1 : -1) {
if (mu[d] != 0) {
int64_t r = ((int64_t)(omega[d]));
if (r <= kmax) {
S[r] = (S[r] + FLOW_CHECKED_DIV((N), ((((int64_t)(d)) * ((int64_t)(d))))));
}
}
}
free(mu);
free(omega);
int64_t qmax = FLOW_CHECKED_DIV((N), (((L0 + 1) * (L0 + 1))));
int32_t __flow_step_13 = 1;
for (int32_t q = 1; (1 <= (qmax + 1)) ? q < (qmax + 1) : q > (qmax + 1); q += (1 <= (qmax + 1)) ? 1 : -1) {
int64_t hi = isqrt_i64(FLOW_CHECKED_DIV((N), (q)));
int64_t lo = isqrt_i64(FLOW_CHECKED_DIV((N), ((q + 1))));
int64_t a = lo;
if (L0 > a) {
a = L0;
}
if (hi <= a) {
continue;
}
int32_t __flow_step_14 = 1;
for (int32_t r = 0; (0 <= (kmax + 1)) ? r < (kmax + 1) : r > (kmax + 1); r += (0 <= (kmax + 1)) ? 1 : -1) {
int64_t cnt = (prefix_count_i64_i64(hi, r) - prefix_count_i64_i64(a, r));
if (cnt != 0) {
S[r] = (S[r] + (q * cnt));
}
}
}
int64_t* comb = (int64_t*)(calloc(((kmax + 1) * (kmax + 1)), 8));
int32_t __flow_step_15 = 1;
for (int32_t n = 0; (0 <= (kmax + 1)) ? n < (kmax + 1) : n > (kmax + 1); n += (0 <= (kmax + 1)) ? 1 : -1) {
comb[((n * (kmax + 1)) + 0)] = 1;
comb[((n * (kmax + 1)) + n)] = 1;
int32_t __flow_step_16 = 1;
for (int32_t r = 1; (1 <= n) ? r < n : r > n; r += (1 <= n) ? 1 : -1) {
comb[((n * (kmax + 1)) + r)] = (comb[((((n - 1) * (kmax + 1)) + r) - 1)] + comb[(((n - 1) * (kmax + 1)) + r)]);
}
}
int64_t* C = (int64_t*)(calloc((kmax + 1), 8));
int32_t __flow_step_17 = 1;
for (int32_t k = 0; (0 <= (kmax + 1)) ? k < (kmax + 1) : k > (kmax + 1); k += (0 <= (kmax + 1)) ? 1 : -1) {
int64_t rev_k = (kmax - k);
int64_t val = S[rev_k];
int32_t __flow_step_18 = 1;
for (int32_t j = (rev_k + 1); ((rev_k + 1) <= (kmax + 1)) ? j < (kmax + 1) : j > (kmax + 1); j += ((rev_k + 1) <= (kmax + 1)) ? 1 : -1) {
val = (val - (C[j] * comb[((j * (kmax + 1)) + rev_k)]));
}
C[rev_k] = val;
}
int64_t ans = 1;
int32_t __flow_step_19 = 1;
for (int32_t k = 0; (0 <= (kmax + 1)) ? k < (kmax + 1) : k > (kmax + 1); k += (0 <= (kmax + 1)) ? 1 : -1) {
if (C[k] != 0) {
int64_t v = FLOW_CHECKED_MOD((C[k]), (MOD));
if (v < 0) {
v = (v + MOD);
}
ans = FLOW_CHECKED_MOD(((ans * v)), (MOD));
}
}
printf("%lld\n", ans);
free(C);
free(comb);
free(S);
free(g_memo_val);
free(g_memo_key);
free(g_primes);
return 0;
}
Generated MLIR
module {
llvm.func @printf(!llvm.ptr, ...) -> i32
llvm.mlir.global internal constant @str_0("%lld\n\00") {addr_space = 0 : i32} : !llvm.array<6 x i8>
func.func @gcd(%arg0: i64, %arg1: i64) -> i64 {
%0 = llvm.mlir.constant(1 : i64) : i64
%1 = llvm.alloca %0 x i64 : (i64) -> !llvm.ptr
llvm.store %arg0, %1 : i64, !llvm.ptr
%2 = llvm.mlir.constant(1 : i64) : i64
%3 = llvm.alloca %2 x i64 : (i64) -> !llvm.ptr
llvm.store %arg1, %3 : i64, !llvm.ptr
cf.br ^bb0
^bb0:
%4 = llvm.load %3 : !llvm.ptr -> i64
%5 = arith.constant 0 : i32
%7 = arith.extsi %5 : i32 to i64
%6 = arith.cmpi ne, %4, %7 : i64
cf.cond_br %6, ^bb1, ^bb2
^bb1:
%8 = llvm.load %1 : !llvm.ptr -> i64
%9 = llvm.load %3 : !llvm.ptr -> i64
%10 = arith.remsi %8, %9 : i64
%11 = llvm.load %3 : !llvm.ptr -> i64
llvm.store %11, %1 : i64, !llvm.ptr
llvm.store %10, %3 : i64, !llvm.ptr
cf.br ^bb0
^bb2:
%12 = llvm.load %1 : !llvm.ptr -> i64
func.return %12 : i64
}
func.func @lcm(%arg0: i64, %arg1: i64) -> i64 {
%13 = arith.constant 0 : i32
%15 = arith.extsi %13 : i32 to i64
%14 = arith.cmpi eq, %arg0, %15 : i64
%16 = scf.if %14 -> (i1) {
%17 = arith.constant true
scf.yield %17 : i1
} else {
%18 = arith.constant 0 : i32
%20 = arith.extsi %18 : i32 to i64
%19 = arith.cmpi eq, %arg1, %20 : i64
scf.yield %19 : i1
}
cf.cond_br %16, ^bb3, ^bb4
^bb3:
%21 = arith.constant 0 : i32
%22 = arith.extsi %21 : i32 to i64
func.return %22 : i64
^bb4:
cf.br ^bb5
^bb5:
%23 = func.call @gcd(%arg0, %arg1) : (i64, i64) -> i64
%24 = arith.divsi %arg0, %23 : i64
%25 = arith.muli %24, %arg1 : i64
func.return %25 : i64
}
func.func @isqrt(%arg0: i64) -> i64 {
%26 = arith.constant 2 : i32
%28 = arith.extsi %26 : i32 to i64
%27 = arith.cmpi slt, %arg0, %28 : i64
cf.cond_br %27, ^bb6, ^bb7
^bb6:
func.return %arg0 : i64
^bb7:
cf.br ^bb8
^bb8:
%29 = llvm.mlir.constant(1 : i64) : i64
%30 = llvm.alloca %29 x i64 : (i64) -> !llvm.ptr
llvm.store %arg0, %30 : i64, !llvm.ptr
%31 = llvm.load %30 : !llvm.ptr -> i64
%32 = arith.constant 1 : i32
%34 = arith.extsi %32 : i32 to i64
%33 = arith.addi %31, %34 : i64
%35 = arith.constant 2 : i32
%37 = arith.extsi %35 : i32 to i64
%36 = arith.divsi %33, %37 : i64
%38 = llvm.mlir.constant(1 : i64) : i64
%39 = llvm.alloca %38 x i64 : (i64) -> !llvm.ptr
llvm.store %36, %39 : i64, !llvm.ptr
cf.br ^bb9
^bb9:
%40 = llvm.load %39 : !llvm.ptr -> i64
%41 = llvm.load %30 : !llvm.ptr -> i64
%42 = arith.cmpi slt, %40, %41 : i64
cf.cond_br %42, ^bb10, ^bb11
^bb10:
%43 = llvm.load %39 : !llvm.ptr -> i64
llvm.store %43, %30 : i64, !llvm.ptr
%44 = llvm.load %30 : !llvm.ptr -> i64
%45 = llvm.load %30 : !llvm.ptr -> i64
%46 = arith.divsi %arg0, %45 : i64
%47 = arith.addi %44, %46 : i64
%48 = arith.constant 2 : i32
%50 = arith.extsi %48 : i32 to i64
%49 = arith.divsi %47, %50 : i64
llvm.store %49, %39 : i64, !llvm.ptr
cf.br ^bb9
^bb11:
%51 = llvm.load %30 : !llvm.ptr -> i64
func.return %51 : i64
}
func.func @mulmod(%arg0: i64, %arg1: i64, %arg2: i64) -> i64 {
%52 = arith.remsi %arg0, %arg2 : i64
%53 = llvm.mlir.constant(1 : i64) : i64
%54 = llvm.alloca %53 x i64 : (i64) -> !llvm.ptr
llvm.store %52, %54 : i64, !llvm.ptr
%55 = arith.remsi %arg1, %arg2 : i64
%56 = llvm.mlir.constant(1 : i64) : i64
%57 = llvm.alloca %56 x i64 : (i64) -> !llvm.ptr
llvm.store %55, %57 : i64, !llvm.ptr
%58 = arith.constant 0 : i32
%59 = arith.extsi %58 : i32 to i64
%60 = llvm.mlir.constant(1 : i64) : i64
%61 = llvm.alloca %60 x i64 : (i64) -> !llvm.ptr
llvm.store %59, %61 : i64, !llvm.ptr
cf.br ^bb12
^bb12:
%62 = llvm.load %57 : !llvm.ptr -> i64
%63 = arith.constant 0 : i32
%65 = arith.extsi %63 : i32 to i64
%64 = arith.cmpi sgt, %62, %65 : i64
cf.cond_br %64, ^bb13, ^bb14
^bb13:
%66 = llvm.load %57 : !llvm.ptr -> i64
%67 = arith.constant 2 : i32
%69 = arith.extsi %67 : i32 to i64
%68 = arith.remsi %66, %69 : i64
%70 = arith.constant 1 : i32
%72 = arith.extsi %70 : i32 to i64
%71 = arith.cmpi eq, %68, %72 : i64
cf.cond_br %71, ^bb15, ^bb16
^bb15:
%73 = llvm.load %61 : !llvm.ptr -> i64
%74 = llvm.load %54 : !llvm.ptr -> i64
%75 = arith.addi %73, %74 : i64
%76 = arith.remsi %75, %arg2 : i64
llvm.store %76, %61 : i64, !llvm.ptr
cf.br ^bb17
^bb16:
cf.br ^bb17
^bb17:
%77 = llvm.load %54 : !llvm.ptr -> i64
%78 = arith.constant 2 : i32
%80 = arith.extsi %78 : i32 to i64
%79 = arith.muli %77, %80 : i64
%81 = arith.remsi %79, %arg2 : i64
llvm.store %81, %54 : i64, !llvm.ptr
%82 = llvm.load %57 : !llvm.ptr -> i64
%83 = arith.constant 2 : i32
%85 = arith.extsi %83 : i32 to i64
%84 = arith.divsi %82, %85 : i64
llvm.store %84, %57 : i64, !llvm.ptr
cf.br ^bb12
^bb14:
%86 = llvm.load %61 : !llvm.ptr -> i64
func.return %86 : i64
}
func.func @mod_pow(%arg0: i64, %arg1: i64, %arg2: i64) -> i64 {
%87 = arith.constant 1 : i32
%89 = arith.extsi %87 : i32 to i64
%88 = arith.cmpi eq, %arg2, %89 : i64
cf.cond_br %88, ^bb18, ^bb19
^bb18:
%90 = arith.constant 0 : i32
%91 = arith.extsi %90 : i32 to i64
func.return %91 : i64
^bb19:
cf.br ^bb20
^bb20:
%92 = arith.constant 1 : i32
%93 = arith.extsi %92 : i32 to i64
%94 = llvm.mlir.constant(1 : i64) : i64
%95 = llvm.alloca %94 x i64 : (i64) -> !llvm.ptr
llvm.store %93, %95 : i64, !llvm.ptr
%96 = arith.remsi %arg0, %arg2 : i64
%97 = llvm.mlir.constant(1 : i64) : i64
%98 = llvm.alloca %97 x i64 : (i64) -> !llvm.ptr
llvm.store %96, %98 : i64, !llvm.ptr
%99 = llvm.mlir.constant(1 : i64) : i64
%100 = llvm.alloca %99 x i64 : (i64) -> !llvm.ptr
llvm.store %arg1, %100 : i64, !llvm.ptr
cf.br ^bb21
^bb21:
%101 = llvm.load %100 : !llvm.ptr -> i64
%102 = arith.constant 0 : i32
%104 = arith.extsi %102 : i32 to i64
%103 = arith.cmpi sgt, %101, %104 : i64
cf.cond_br %103, ^bb22, ^bb23
^bb22:
%105 = llvm.load %100 : !llvm.ptr -> i64
%106 = arith.constant 2 : i32
%108 = arith.extsi %106 : i32 to i64
%107 = arith.remsi %105, %108 : i64
%109 = arith.constant 1 : i32
%111 = arith.extsi %109 : i32 to i64
%110 = arith.cmpi eq, %107, %111 : i64
cf.cond_br %110, ^bb24, ^bb25
^bb24:
%113 = llvm.load %95 : !llvm.ptr -> i64
%114 = llvm.load %98 : !llvm.ptr -> i64
%112 = func.call @mulmod(%113, %114, %arg2) : (i64, i64, i64) -> i64
llvm.store %112, %95 : i64, !llvm.ptr
cf.br ^bb26
^bb25:
cf.br ^bb26
^bb26:
%116 = llvm.load %98 : !llvm.ptr -> i64
%117 = llvm.load %98 : !llvm.ptr -> i64
%115 = func.call @mulmod(%116, %117, %arg2) : (i64, i64, i64) -> i64
llvm.store %115, %98 : i64, !llvm.ptr
%118 = llvm.load %100 : !llvm.ptr -> i64
%119 = arith.constant 2 : i32
%121 = arith.extsi %119 : i32 to i64
%120 = arith.divsi %118, %121 : i64
llvm.store %120, %100 : i64, !llvm.ptr
cf.br ^bb21
^bb23:
%122 = llvm.load %95 : !llvm.ptr -> i64
func.return %122 : i64
}
func.func @is_prime(%arg0: i64) -> i1 {
%123 = arith.constant 2 : i32
%125 = arith.extsi %123 : i32 to i64
%124 = arith.cmpi slt, %arg0, %125 : i64
cf.cond_br %124, ^bb27, ^bb28
^bb27:
%126 = arith.constant 0 : i1
func.return %126 : i1
^bb28:
cf.br ^bb29
^bb29:
%127 = arith.constant 4 : i32
%129 = arith.extsi %127 : i32 to i64
%128 = arith.cmpi slt, %arg0, %129 : i64
cf.cond_br %128, ^bb30, ^bb31
^bb30:
%130 = arith.constant 1 : i1
func.return %130 : i1
^bb31:
cf.br ^bb32
^bb32:
%131 = arith.constant 2 : i32
%133 = arith.extsi %131 : i32 to i64
%132 = arith.remsi %arg0, %133 : i64
%134 = arith.constant 0 : i32
%136 = arith.extsi %134 : i32 to i64
%135 = arith.cmpi eq, %132, %136 : i64
%137 = scf.if %135 -> (i1) {
%138 = arith.constant true
scf.yield %138 : i1
} else {
%139 = arith.constant 3 : i32
%141 = arith.extsi %139 : i32 to i64
%140 = arith.remsi %arg0, %141 : i64
%142 = arith.constant 0 : i32
%144 = arith.extsi %142 : i32 to i64
%143 = arith.cmpi eq, %140, %144 : i64
scf.yield %143 : i1
}
cf.cond_br %137, ^bb33, ^bb34
^bb33:
%145 = arith.constant 0 : i1
func.return %145 : i1
^bb34:
cf.br ^bb35
^bb35:
%146 = arith.constant 5 : i32
%147 = arith.extsi %146 : i32 to i64
%148 = llvm.mlir.constant(1 : i64) : i64
%149 = llvm.alloca %148 x i64 : (i64) -> !llvm.ptr
llvm.store %147, %149 : i64, !llvm.ptr
cf.br ^bb36
^bb36:
%150 = llvm.load %149 : !llvm.ptr -> i64
%151 = llvm.load %149 : !llvm.ptr -> i64
%152 = arith.muli %150, %151 : i64
%153 = arith.cmpi sle, %152, %arg0 : i64
cf.cond_br %153, ^bb37, ^bb38
^bb37:
%154 = llvm.load %149 : !llvm.ptr -> i64
%155 = arith.remsi %arg0, %154 : i64
%156 = arith.constant 0 : i32
%158 = arith.extsi %156 : i32 to i64
%157 = arith.cmpi eq, %155, %158 : i64
%159 = scf.if %157 -> (i1) {
%160 = arith.constant true
scf.yield %160 : i1
} else {
%161 = llvm.load %149 : !llvm.ptr -> i64
%162 = arith.constant 2 : i32
%164 = arith.extsi %162 : i32 to i64
%163 = arith.addi %161, %164 : i64
%165 = arith.remsi %arg0, %163 : i64
%166 = arith.constant 0 : i32
%168 = arith.extsi %166 : i32 to i64
%167 = arith.cmpi eq, %165, %168 : i64
scf.yield %167 : i1
}
cf.cond_br %159, ^bb39, ^bb40
^bb39:
%169 = arith.constant 0 : i1
func.return %169 : i1
^bb40:
cf.br ^bb41
^bb41:
%170 = llvm.load %149 : !llvm.ptr -> i64
%171 = arith.constant 6 : i32
%173 = arith.extsi %171 : i32 to i64
%172 = arith.addi %170, %173 : i64
llvm.store %172, %149 : i64, !llvm.ptr
cf.br ^bb36
^bb38:
%174 = arith.constant 1 : i1
func.return %174 : i1
}
func.func private @calloc(i64, i64) -> !llvm.ptr
func.func private @free(!llvm.ptr) -> ()
func.func private @memset(!llvm.ptr, i64, i64) -> !llvm.ptr
// Constant: MOD
llvm.mlir.global internal constant @MOD(1000000007 : i64) : i64
// Module static: g_primes
llvm.mlir.global internal @g_primes() {addr_space = 0 : i32} : !llvm.ptr {
%175 = llvm.mlir.zero : !llvm.ptr
llvm.return %175 : !llvm.ptr
}
// Module static: g_nprimes
llvm.mlir.global internal @g_nprimes(0 : i64) : i64
func.func @pi(%arg0: i64) -> i64 {
%176 = arith.constant 0 : i32
%177 = arith.extsi %176 : i32 to i64
%178 = llvm.mlir.constant(1 : i64) : i64
%179 = llvm.alloca %178 x i64 : (i64) -> !llvm.ptr
llvm.store %177, %179 : i64, !llvm.ptr
%180 = llvm.mlir.addressof @g_nprimes : !llvm.ptr
%181 = llvm.load %180 : !llvm.ptr -> i64
%182 = llvm.mlir.constant(1 : i64) : i64
%183 = llvm.alloca %182 x i64 : (i64) -> !llvm.ptr
llvm.store %181, %183 : i64, !llvm.ptr
cf.br ^bb42
^bb42:
%184 = llvm.load %179 : !llvm.ptr -> i64
%185 = llvm.load %183 : !llvm.ptr -> i64
%186 = arith.cmpi slt, %184, %185 : i64
cf.cond_br %186, ^bb43, ^bb44
^bb43:
%187 = llvm.load %179 : !llvm.ptr -> i64
%188 = llvm.load %183 : !llvm.ptr -> i64
%189 = arith.addi %187, %188 : i64
%190 = arith.constant 2 : i32
%192 = arith.extsi %190 : i32 to i64
%191 = arith.divsi %189, %192 : i64
%194 = llvm.mlir.addressof @g_primes : !llvm.ptr
%195 = llvm.load %194 : !llvm.ptr -> !llvm.ptr
%196 = llvm.getelementptr %195[%191] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%193 = llvm.load %196 : !llvm.ptr -> i64
%197 = arith.cmpi sle, %193, %arg0 : i64
cf.cond_br %197, ^bb45, ^bb46
^bb45:
%198 = arith.constant 1 : i32
%200 = arith.extsi %198 : i32 to i64
%199 = arith.addi %191, %200 : i64
llvm.store %199, %179 : i64, !llvm.ptr
cf.br ^bb47
^bb46:
llvm.store %191, %183 : i64, !llvm.ptr
cf.br ^bb47
^bb47:
cf.br ^bb42
^bb44:
%201 = llvm.load %179 : !llvm.ptr -> i64
func.return %201 : i64
}
func.func @kth_root_floor(%arg0: i64, %arg1: i64) -> i64 {
%202 = arith.constant 2 : i32
%204 = arith.extsi %202 : i32 to i64
%203 = arith.cmpi eq, %arg1, %204 : i64
cf.cond_br %203, ^bb48, ^bb49
^bb48:
%205 = func.call @isqrt(%arg0) : (i64) -> i64
func.return %205 : i64
^bb49:
cf.br ^bb50
^bb50:
%206 = arith.constant 1 : i32
%207 = arith.extsi %206 : i32 to i64
%208 = llvm.mlir.constant(1 : i64) : i64
%209 = llvm.alloca %208 x i64 : (i64) -> !llvm.ptr
llvm.store %207, %209 : i64, !llvm.ptr
%210 = arith.constant 2 : i32
%211 = arith.extsi %210 : i32 to i64
%212 = llvm.mlir.constant(1 : i64) : i64
%213 = llvm.alloca %212 x i64 : (i64) -> !llvm.ptr
llvm.store %211, %213 : i64, !llvm.ptr
cf.br ^bb51
^bb51:
%214 = llvm.load %213 : !llvm.ptr -> i64
%215 = arith.constant 0 : i32
%217 = arith.extsi %215 : i32 to i64
%216 = arith.cmpi sgt, %214, %217 : i64
cf.cond_br %216, ^bb52, ^bb53
^bb52:
%218 = arith.constant 1 : i32
%219 = arith.extsi %218 : i32 to i64
%220 = llvm.mlir.constant(1 : i64) : i64
%221 = llvm.alloca %220 x i64 : (i64) -> !llvm.ptr
llvm.store %219, %221 : i64, !llvm.ptr
%222 = arith.constant 1 : i32
%223 = arith.extsi %222 : i32 to i64
%224 = llvm.mlir.constant(1 : i64) : i64
%225 = llvm.alloca %224 x i64 : (i64) -> !llvm.ptr
llvm.store %223, %225 : i64, !llvm.ptr
%226 = arith.constant 0 : i32
%227 = arith.index_cast %226 : i32 to index
%228 = arith.index_cast %arg1 : i32 to index
%230 = arith.constant 1 : index
%231 = arith.constant -1 : index
%232 = arith.cmpi sle, %227, %228 : index
%229 = arith.select %232, %230, %231 : index
cf.br ^bb54(%227 : index)
^bb54(%233: index):
%234 = arith.cmpi slt, %233, %228 : index
%235 = arith.cmpi sgt, %233, %228 : index
%236 = arith.select %232, %234, %235 : i1
cf.cond_br %236, ^bb55(%233 : index), ^bb56(%233 : index)
^bb55(%237: index):
%238 = llvm.load %221 : !llvm.ptr -> i64
%239 = llvm.load %213 : !llvm.ptr -> i64
%240 = arith.divsi %arg0, %239 : i64
%241 = arith.cmpi sgt, %238, %240 : i64
cf.cond_br %241, ^bb57, ^bb58
^bb57:
%242 = arith.constant 0 : i32
%243 = arith.extsi %242 : i32 to i64
llvm.store %243, %225 : i64, !llvm.ptr
cf.br ^bb59
^bb58:
cf.br ^bb59
^bb59:
%244 = llvm.load %225 : !llvm.ptr -> i64
%245 = arith.constant 1 : i32
%247 = arith.extsi %245 : i32 to i64
%246 = arith.cmpi eq, %244, %247 : i64
cf.cond_br %246, ^bb60, ^bb61
^bb60:
%248 = llvm.load %221 : !llvm.ptr -> i64
%249 = llvm.load %213 : !llvm.ptr -> i64
%250 = arith.muli %248, %249 : i64
llvm.store %250, %221 : i64, !llvm.ptr
cf.br ^bb62
^bb61:
cf.br ^bb62
^bb62:
%251 = arith.addi %237, %229 : index
cf.br ^bb54(%251 : index)
^bb56(%252: index):
%253 = llvm.load %225 : !llvm.ptr -> i64
%254 = arith.constant 0 : i32
%256 = arith.extsi %254 : i32 to i64
%255 = arith.cmpi eq, %253, %256 : i64
cf.cond_br %255, ^bb63, ^bb64
^bb63:
cf.br ^bb53
^bb64:
cf.br ^bb65
^bb65:
%257 = llvm.load %213 : !llvm.ptr -> i64
%258 = arith.constant 2 : i32
%260 = arith.extsi %258 : i32 to i64
%259 = arith.muli %257, %260 : i64
llvm.store %259, %213 : i64, !llvm.ptr
%261 = llvm.load %213 : !llvm.ptr -> i64
%262 = arith.constant 1000000000 : i32
%264 = arith.extsi %262 : i32 to i64
%263 = arith.cmpi sgt, %261, %264 : i64
cf.cond_br %263, ^bb66, ^bb67
^bb66:
cf.br ^bb53
^bb67:
cf.br ^bb68
^bb68:
cf.br ^bb51
^bb53:
cf.br ^bb69
^bb69:
%265 = llvm.load %209 : !llvm.ptr -> i64
%266 = arith.constant 1 : i32
%268 = arith.extsi %266 : i32 to i64
%267 = arith.addi %265, %268 : i64
%269 = llvm.load %213 : !llvm.ptr -> i64
%270 = arith.cmpi slt, %267, %269 : i64
cf.cond_br %270, ^bb70, ^bb71
^bb70:
%271 = llvm.load %209 : !llvm.ptr -> i64
%272 = llvm.load %213 : !llvm.ptr -> i64
%273 = arith.addi %271, %272 : i64
%274 = arith.constant 2 : i32
%276 = arith.extsi %274 : i32 to i64
%275 = arith.divsi %273, %276 : i64
%277 = arith.constant 1 : i32
%278 = arith.extsi %277 : i32 to i64
%279 = llvm.mlir.constant(1 : i64) : i64
%280 = llvm.alloca %279 x i64 : (i64) -> !llvm.ptr
llvm.store %278, %280 : i64, !llvm.ptr
%281 = arith.constant 1 : i32
%282 = arith.extsi %281 : i32 to i64
%283 = llvm.mlir.constant(1 : i64) : i64
%284 = llvm.alloca %283 x i64 : (i64) -> !llvm.ptr
llvm.store %282, %284 : i64, !llvm.ptr
%285 = arith.constant 0 : i32
%286 = arith.index_cast %285 : i32 to index
%287 = arith.index_cast %arg1 : i32 to index
%289 = arith.constant 1 : index
%290 = arith.constant -1 : index
%291 = arith.cmpi sle, %286, %287 : index
%288 = arith.select %291, %289, %290 : index
cf.br ^bb72(%286 : index)
^bb72(%292: index):
%293 = arith.cmpi slt, %292, %287 : index
%294 = arith.cmpi sgt, %292, %287 : index
%295 = arith.select %291, %293, %294 : i1
cf.cond_br %295, ^bb73(%292 : index), ^bb74(%292 : index)
^bb73(%296: index):
%297 = llvm.load %280 : !llvm.ptr -> i64
%298 = arith.divsi %arg0, %275 : i64
%299 = arith.cmpi sgt, %297, %298 : i64
cf.cond_br %299, ^bb75, ^bb76
^bb75:
%300 = arith.constant 0 : i32
%301 = arith.extsi %300 : i32 to i64
llvm.store %301, %284 : i64, !llvm.ptr
cf.br ^bb77
^bb76:
cf.br ^bb77
^bb77:
%302 = llvm.load %284 : !llvm.ptr -> i64
%303 = arith.constant 1 : i32
%305 = arith.extsi %303 : i32 to i64
%304 = arith.cmpi eq, %302, %305 : i64
cf.cond_br %304, ^bb78, ^bb79
^bb78:
%306 = llvm.load %280 : !llvm.ptr -> i64
%307 = arith.muli %306, %275 : i64
llvm.store %307, %280 : i64, !llvm.ptr
cf.br ^bb80
^bb79:
cf.br ^bb80
^bb80:
%308 = arith.addi %296, %288 : index
cf.br ^bb72(%308 : index)
^bb74(%309: index):
%310 = llvm.load %284 : !llvm.ptr -> i64
%311 = arith.constant 1 : i32
%313 = arith.extsi %311 : i32 to i64
%312 = arith.cmpi eq, %310, %313 : i64
cf.cond_br %312, ^bb81, ^bb82
^bb81:
llvm.store %275, %209 : i64, !llvm.ptr
cf.br ^bb83
^bb82:
llvm.store %275, %213 : i64, !llvm.ptr
cf.br ^bb83
^bb83:
cf.br ^bb69
^bb71:
%314 = llvm.load %209 : !llvm.ptr -> i64
func.return %314 : i64
}
// Constant: MEMO_SIZE
llvm.mlir.global internal constant @MEMO_SIZE(4000007 : i64) : i64
// Module static: g_memo_key
llvm.mlir.global internal @g_memo_key() {addr_space = 0 : i32} : !llvm.ptr {
%315 = llvm.mlir.zero : !llvm.ptr
llvm.return %315 : !llvm.ptr
}
// Module static: g_memo_val
llvm.mlir.global internal @g_memo_val() {addr_space = 0 : i32} : !llvm.ptr {
%316 = llvm.mlir.zero : !llvm.ptr
llvm.return %316 : !llvm.ptr
}
func.func @memo_key(%arg0: i64, %arg1: i64, %arg2: i64) -> i64 {
%317 = arith.constant 1000 : i32
%319 = arith.extsi %317 : i32 to i64
%318 = arith.muli %arg0, %319 : i64
%320 = arith.constant 100 : i32
%322 = arith.extsi %320 : i32 to i64
%321 = arith.muli %arg1, %322 : i64
%323 = arith.addi %318, %321 : i64
%324 = arith.addi %323, %arg2 : i64
func.return %324 : i64
}
func.func @count_sqfree(%arg0: i64, %arg1: i64, %arg2: i64) -> i64 {
%325 = arith.constant 0 : i32
%327 = arith.extsi %325 : i32 to i64
%326 = arith.cmpi eq, %arg1, %327 : i64
cf.cond_br %326, ^bb84, ^bb85
^bb84:
%328 = arith.constant 1 : i32
%329 = arith.extsi %328 : i32 to i64
func.return %329 : i64
^bb85:
cf.br ^bb86
^bb86:
%330 = arith.constant 1 : i32
%332 = arith.extsi %330 : i32 to i64
%331 = arith.cmpi eq, %arg1, %332 : i64
cf.cond_br %331, ^bb87, ^bb88
^bb87:
%333 = func.call @pi(%arg0) : (i64) -> i64
%334 = arith.subi %333, %arg2 : i64
func.return %334 : i64
^bb88:
cf.br ^bb89
^bb89:
%335 = func.call @kth_root_floor(%arg0, %arg1) : (i64, i64) -> i64
%336 = func.call @pi(%335) : (i64) -> i64
%337 = arith.cmpi sle, %336, %arg2 : i64
cf.cond_br %337, ^bb90, ^bb91
^bb90:
%338 = arith.constant 0 : i32
%339 = arith.extsi %338 : i32 to i64
func.return %339 : i64
^bb91:
cf.br ^bb92
^bb92:
%340 = func.call @memo_key(%arg0, %arg1, %arg2) : (i64, i64, i64) -> i64
%341 = llvm.mlir.addressof @MEMO_SIZE : !llvm.ptr
%342 = llvm.load %341 : !llvm.ptr -> i64
%343 = arith.remsi %340, %342 : i64
%345 = llvm.mlir.addressof @g_memo_key : !llvm.ptr
%346 = llvm.load %345 : !llvm.ptr -> !llvm.ptr
%347 = llvm.getelementptr %346[%343] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%344 = llvm.load %347 : !llvm.ptr -> i64
%348 = arith.cmpi eq, %344, %340 : i64
cf.cond_br %348, ^bb93, ^bb94
^bb93:
%350 = llvm.mlir.addressof @g_memo_val : !llvm.ptr
%351 = llvm.load %350 : !llvm.ptr -> !llvm.ptr
%352 = llvm.getelementptr %351[%343] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%349 = llvm.load %352 : !llvm.ptr -> i64
func.return %349 : i64
^bb94:
cf.br ^bb95
^bb95:
%353 = arith.constant 0 : i32
%354 = arith.extsi %353 : i32 to i64
%355 = llvm.mlir.constant(1 : i64) : i64
%356 = llvm.alloca %355 x i64 : (i64) -> !llvm.ptr
llvm.store %354, %356 : i64, !llvm.ptr
%357 = arith.index_cast %arg2 : i32 to index
%358 = arith.index_cast %336 : i32 to index
%360 = arith.constant 1 : index
%361 = arith.constant -1 : index
%362 = arith.cmpi sle, %357, %358 : index
%359 = arith.select %362, %360, %361 : index
cf.br ^bb96(%357 : index)
^bb96(%363: index):
%364 = arith.cmpi slt, %363, %358 : index
%365 = arith.cmpi sgt, %363, %358 : index
%366 = arith.select %362, %364, %365 : i1
cf.cond_br %366, ^bb97(%363 : index), ^bb98(%363 : index)
^bb97(%367: index):
%369 = llvm.mlir.addressof @g_primes : !llvm.ptr
%370 = llvm.load %369 : !llvm.ptr -> !llvm.ptr
%371 = arith.index_cast %367 : index to i64
%372 = llvm.getelementptr %370[%371] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%368 = llvm.load %372 : !llvm.ptr -> i64
%373 = llvm.load %356 : !llvm.ptr -> i64
%375 = arith.divsi %arg0, %368 : i64
%376 = arith.constant 1 : i32
%378 = arith.extsi %376 : i32 to i64
%377 = arith.subi %arg1, %378 : i64
%379 = arith.constant 1 : i32
%381 = arith.index_cast %367 : index to i32
%380 = arith.addi %381, %379 : i32
%382 = arith.extsi %380 : i32 to i64
%374 = func.call @count_sqfree(%375, %377, %382) : (i64, i64, i64) -> i64
%383 = arith.addi %373, %374 : i64
llvm.store %383, %356 : i64, !llvm.ptr
%384 = arith.addi %367, %359 : index
cf.br ^bb96(%384 : index)
^bb98(%385: index):
%386 = llvm.mlir.addressof @g_memo_key : !llvm.ptr
%387 = llvm.load %386 : !llvm.ptr -> !llvm.ptr
%388 = llvm.getelementptr %387[%343] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %340, %388 : i64, !llvm.ptr
%389 = llvm.load %356 : !llvm.ptr -> i64
%390 = llvm.mlir.addressof @g_memo_val : !llvm.ptr
%391 = llvm.load %390 : !llvm.ptr -> !llvm.ptr
%392 = llvm.getelementptr %391[%343] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %389, %392 : i64, !llvm.ptr
%393 = llvm.load %356 : !llvm.ptr -> i64
func.return %393 : i64
}
func.func @prefix_count(%arg0: i64, %arg1: i64) -> i64 {
%394 = arith.constant 0 : i32
%396 = arith.extsi %394 : i32 to i64
%395 = arith.cmpi eq, %arg1, %396 : i64
cf.cond_br %395, ^bb99, ^bb100
^bb99:
%397 = arith.constant 1 : i32
%399 = arith.extsi %397 : i32 to i64
%398 = arith.cmpi sge, %arg0, %399 : i64
cf.cond_br %398, ^bb102, ^bb103
^bb102:
%400 = arith.constant 1 : i32
%401 = arith.extsi %400 : i32 to i64
func.return %401 : i64
^bb103:
cf.br ^bb104
^bb104:
%402 = arith.constant 0 : i32
%403 = arith.extsi %402 : i32 to i64
func.return %403 : i64
^bb100:
cf.br ^bb101
^bb101:
%405 = arith.constant 0 : i32
%406 = arith.extsi %405 : i32 to i64
%404 = func.call @count_sqfree(%arg0, %arg1, %406) : (i64, i64, i64) -> i64
func.return %404 : i64
}
func.func @main() -> i32 {
%407 = arith.constant 9999995705032704 : i32
%408 = arith.extsi %407 : i32 to i64
%409 = func.call @isqrt(%408) : (i64) -> i64
%410 = arith.constant 1 : i32
%412 = arith.extsi %410 : i32 to i64
%411 = arith.addi %409, %412 : i64
%413 = arith.constant 2 : i32
%415 = arith.extsi %413 : i32 to i64
%414 = arith.divsi %411, %415 : i64
%417 = arith.constant 1 : i32
%418 = arith.extsi %417 : i32 to i64
%416 = func.call @calloc(%414, %418) : (i64, i64) -> !llvm.ptr
%419 = arith.constant 0 : i32
%420 = arith.index_cast %419 : i32 to index
%421 = arith.index_cast %414 : i32 to index
%423 = arith.constant 1 : index
%424 = arith.constant -1 : index
%425 = arith.cmpi sle, %420, %421 : index
%422 = arith.select %425, %423, %424 : index
cf.br ^bb105(%420 : index)
^bb105(%426: index):
%427 = arith.cmpi slt, %426, %421 : index
%428 = arith.cmpi sgt, %426, %421 : index
%429 = arith.select %425, %427, %428 : i1
cf.cond_br %429, ^bb106(%426 : index), ^bb107(%426 : index)
^bb106(%430: index):
%431 = arith.constant 1 : i32
%432 = arith.trunci %431 : i32 to i8
%433 = arith.index_cast %430 : index to i64
%434 = llvm.getelementptr %416[%433] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %432, %434 : i8, !llvm.ptr
%435 = arith.addi %430, %422 : index
cf.br ^bb105(%435 : index)
^bb107(%436: index):
%437 = arith.constant 0 : i32
%438 = arith.constant 0 : i32
%439 = arith.trunci %437 : i32 to i8
%440 = arith.extsi %438 : i32 to i64
%441 = llvm.getelementptr %416[%440] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %439, %441 : i8, !llvm.ptr
%442 = func.call @isqrt(%409) : (i64) -> i64
%443 = arith.constant 1 : i32
%444 = arith.constant 2 : i32
%446 = arith.extsi %444 : i32 to i64
%445 = arith.divsi %442, %446 : i64
%447 = arith.constant 1 : i32
%449 = arith.extsi %447 : i32 to i64
%448 = arith.addi %445, %449 : i64
%450 = arith.index_cast %443 : i32 to index
%451 = arith.index_cast %448 : i32 to index
%453 = arith.constant 1 : index
%454 = arith.constant -1 : index
%455 = arith.cmpi sle, %450, %451 : index
%452 = arith.select %455, %453, %454 : index
cf.br ^bb108(%450 : index)
^bb108(%456: index):
%457 = arith.cmpi slt, %456, %451 : index
%458 = arith.cmpi sgt, %456, %451 : index
%459 = arith.select %455, %457, %458 : i1
cf.cond_br %459, ^bb109(%456 : index), ^bb110(%456 : index)
^bb109(%460: index):
%462 = arith.index_cast %460 : index to i64
%463 = llvm.getelementptr %416[%462] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%461 = llvm.load %463 : !llvm.ptr -> i8
%464 = arith.constant 1 : i32
%466 = arith.extsi %461 : i8 to i32
%465 = arith.cmpi eq, %466, %464 : i32
cf.cond_br %465, ^bb111, ^bb112
^bb111:
%467 = arith.constant 2 : i32
%469 = arith.index_cast %460 : index to i32
%468 = arith.muli %467, %469 : i32
%470 = arith.constant 1 : i32
%471 = arith.addi %468, %470 : i32
%472 = arith.extsi %471 : i32 to i64
%473 = arith.muli %472, %472 : i64
%474 = arith.constant 2 : i32
%476 = arith.extsi %474 : i32 to i64
%475 = arith.divsi %473, %476 : i64
%477 = llvm.mlir.constant(1 : i64) : i64
%478 = llvm.alloca %477 x i64 : (i64) -> !llvm.ptr
llvm.store %475, %478 : i64, !llvm.ptr
cf.br ^bb114
^bb114:
%479 = llvm.load %478 : !llvm.ptr -> i64
%480 = arith.cmpi slt, %479, %414 : i64
cf.cond_br %480, ^bb115, ^bb116
^bb115:
%481 = arith.constant 0 : i32
%482 = llvm.load %478 : !llvm.ptr -> i64
%483 = arith.trunci %481 : i32 to i8
%484 = llvm.getelementptr %416[%482] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %483, %484 : i8, !llvm.ptr
%485 = llvm.load %478 : !llvm.ptr -> i64
%486 = arith.addi %485, %472 : i64
llvm.store %486, %478 : i64, !llvm.ptr
cf.br ^bb114
^bb116:
cf.br ^bb113
^bb112:
cf.br ^bb113
^bb113:
%487 = arith.addi %460, %452 : index
cf.br ^bb108(%487 : index)
^bb110(%488: index):
%489 = arith.constant 1 : i32
%490 = arith.extsi %489 : i32 to i64
%491 = llvm.mlir.constant(1 : i64) : i64
%492 = llvm.alloca %491 x i64 : (i64) -> !llvm.ptr
llvm.store %490, %492 : i64, !llvm.ptr
%493 = arith.constant 1 : i32
%494 = arith.index_cast %493 : i32 to index
%495 = arith.index_cast %414 : i32 to index
%497 = arith.constant 1 : index
%498 = arith.constant -1 : index
%499 = arith.cmpi sle, %494, %495 : index
%496 = arith.select %499, %497, %498 : index
cf.br ^bb117(%494 : index)
^bb117(%500: index):
%501 = arith.cmpi slt, %500, %495 : index
%502 = arith.cmpi sgt, %500, %495 : index
%503 = arith.select %499, %501, %502 : i1
cf.cond_br %503, ^bb118(%500 : index), ^bb119(%500 : index)
^bb118(%504: index):
%506 = arith.index_cast %504 : index to i64
%507 = llvm.getelementptr %416[%506] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%505 = llvm.load %507 : !llvm.ptr -> i8
%508 = arith.constant 1 : i32
%510 = arith.extsi %505 : i8 to i32
%509 = arith.cmpi eq, %510, %508 : i32
cf.cond_br %509, ^bb120, ^bb121
^bb120:
%511 = llvm.load %492 : !llvm.ptr -> i64
%512 = arith.constant 1 : i32
%514 = arith.extsi %512 : i32 to i64
%513 = arith.addi %511, %514 : i64
llvm.store %513, %492 : i64, !llvm.ptr
cf.br ^bb122
^bb121:
cf.br ^bb122
^bb122:
%515 = arith.addi %504, %496 : index
cf.br ^bb117(%515 : index)
^bb119(%516: index):
%518 = llvm.load %492 : !llvm.ptr -> i64
%519 = arith.constant 8 : i32
%520 = arith.extsi %519 : i32 to i64
%517 = func.call @calloc(%518, %520) : (i64, i64) -> !llvm.ptr
%521 = llvm.mlir.addressof @g_primes : !llvm.ptr
llvm.store %517, %521 : !llvm.ptr, !llvm.ptr
%522 = arith.constant 2 : i32
%523 = llvm.mlir.addressof @g_primes : !llvm.ptr
%524 = llvm.load %523 : !llvm.ptr -> !llvm.ptr
%525 = arith.constant 0 : i32
%526 = arith.extsi %522 : i32 to i64
%527 = arith.extsi %525 : i32 to i64
%528 = llvm.getelementptr %524[%527] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %526, %528 : i64, !llvm.ptr
%529 = arith.constant 1 : i32
%530 = arith.extsi %529 : i32 to i64
%531 = llvm.mlir.constant(1 : i64) : i64
%532 = llvm.alloca %531 x i64 : (i64) -> !llvm.ptr
llvm.store %530, %532 : i64, !llvm.ptr
%533 = arith.constant 1 : i32
%534 = arith.index_cast %533 : i32 to index
%535 = arith.index_cast %414 : i32 to index
%537 = arith.constant 1 : index
%538 = arith.constant -1 : index
%539 = arith.cmpi sle, %534, %535 : index
%536 = arith.select %539, %537, %538 : index
cf.br ^bb123(%534 : index)
^bb123(%540: index):
%541 = arith.cmpi slt, %540, %535 : index
%542 = arith.cmpi sgt, %540, %535 : index
%543 = arith.select %539, %541, %542 : i1
cf.cond_br %543, ^bb124(%540 : index), ^bb125(%540 : index)
^bb124(%544: index):
%546 = arith.index_cast %544 : index to i64
%547 = llvm.getelementptr %416[%546] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%545 = llvm.load %547 : !llvm.ptr -> i8
%548 = arith.constant 1 : i32
%550 = arith.extsi %545 : i8 to i32
%549 = arith.cmpi eq, %550, %548 : i32
cf.cond_br %549, ^bb126, ^bb127
^bb126:
%551 = arith.constant 2 : i32
%553 = arith.index_cast %544 : index to i32
%552 = arith.muli %551, %553 : i32
%554 = arith.constant 1 : i32
%555 = arith.addi %552, %554 : i32
%556 = llvm.mlir.addressof @g_primes : !llvm.ptr
%557 = llvm.load %556 : !llvm.ptr -> !llvm.ptr
%558 = llvm.load %532 : !llvm.ptr -> i64
%559 = arith.extsi %555 : i32 to i64
%560 = llvm.getelementptr %557[%558] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %559, %560 : i64, !llvm.ptr
%561 = llvm.load %532 : !llvm.ptr -> i64
%562 = arith.constant 1 : i32
%564 = arith.extsi %562 : i32 to i64
%563 = arith.addi %561, %564 : i64
llvm.store %563, %532 : i64, !llvm.ptr
cf.br ^bb128
^bb127:
cf.br ^bb128
^bb128:
%565 = arith.addi %544, %536 : index
cf.br ^bb123(%565 : index)
^bb125(%566: index):
%567 = llvm.load %492 : !llvm.ptr -> i64
%568 = llvm.mlir.addressof @g_nprimes : !llvm.ptr
llvm.store %567, %568 : i64, !llvm.ptr
func.call @free(%416) : (!llvm.ptr) -> ()
%571 = llvm.mlir.addressof @MEMO_SIZE : !llvm.ptr
%572 = llvm.load %571 : !llvm.ptr -> i64
%573 = arith.constant 8 : i32
%574 = arith.extsi %573 : i32 to i64
%570 = func.call @calloc(%572, %574) : (i64, i64) -> !llvm.ptr
%575 = llvm.mlir.addressof @g_memo_key : !llvm.ptr
llvm.store %570, %575 : !llvm.ptr, !llvm.ptr
%577 = llvm.mlir.addressof @MEMO_SIZE : !llvm.ptr
%578 = llvm.load %577 : !llvm.ptr -> i64
%579 = arith.constant 8 : i32
%580 = arith.extsi %579 : i32 to i64
%576 = func.call @calloc(%578, %580) : (i64, i64) -> !llvm.ptr
%581 = llvm.mlir.addressof @g_memo_val : !llvm.ptr
llvm.store %576, %581 : !llvm.ptr, !llvm.ptr
%582 = arith.constant 0 : i32
%583 = arith.extsi %582 : i32 to i64
%584 = llvm.mlir.constant(1 : i64) : i64
%585 = llvm.alloca %584 x i64 : (i64) -> !llvm.ptr
llvm.store %583, %585 : i64, !llvm.ptr
%586 = arith.constant 0 : i32
%587 = llvm.load %492 : !llvm.ptr -> i64
%588 = arith.index_cast %586 : i32 to index
%589 = arith.index_cast %587 : i32 to index
%591 = arith.constant 1 : index
%592 = arith.constant -1 : index
%593 = arith.cmpi sle, %588, %589 : index
%590 = arith.select %593, %591, %592 : index
cf.br ^bb129(%588 : index)
^bb129(%594: index):
%595 = arith.cmpi slt, %594, %589 : index
%596 = arith.cmpi sgt, %594, %589 : index
%597 = arith.select %593, %595, %596 : i1
cf.cond_br %597, ^bb130(%594 : index), ^bb131(%594 : index)
^bb130(%598: index):
%600 = llvm.mlir.addressof @g_primes : !llvm.ptr
%601 = llvm.load %600 : !llvm.ptr -> !llvm.ptr
%602 = arith.index_cast %598 : index to i64
%603 = llvm.getelementptr %601[%602] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%599 = llvm.load %603 : !llvm.ptr -> i64
%604 = arith.constant 10000 : i32
%606 = arith.extsi %604 : i32 to i64
%605 = arith.cmpi sle, %599, %606 : i64
cf.cond_br %605, ^bb132, ^bb133
^bb132:
%607 = llvm.load %585 : !llvm.ptr -> i64
%608 = arith.constant 1 : i32
%610 = arith.extsi %608 : i32 to i64
%609 = arith.addi %607, %610 : i64
llvm.store %609, %585 : i64, !llvm.ptr
cf.br ^bb134
^bb133:
cf.br ^bb131(%598 : index)
^bb134:
%611 = arith.addi %598, %590 : index
cf.br ^bb129(%611 : index)
^bb131(%612: index):
%613 = arith.constant 1 : i32
%614 = arith.extsi %613 : i32 to i64
%615 = llvm.mlir.constant(1 : i64) : i64
%616 = llvm.alloca %615 x i64 : (i64) -> !llvm.ptr
llvm.store %614, %616 : i64, !llvm.ptr
%617 = arith.constant 0 : i32
%618 = arith.extsi %617 : i32 to i64
%619 = llvm.mlir.constant(1 : i64) : i64
%620 = llvm.alloca %619 x i64 : (i64) -> !llvm.ptr
llvm.store %618, %620 : i64, !llvm.ptr
%621 = arith.constant 0 : i32
%622 = llvm.load %585 : !llvm.ptr -> i64
%623 = arith.index_cast %621 : i32 to index
%624 = arith.index_cast %622 : i32 to index
%626 = arith.constant 1 : index
%627 = arith.constant -1 : index
%628 = arith.cmpi sle, %623, %624 : index
%625 = arith.select %628, %626, %627 : index
cf.br ^bb135(%623 : index)
^bb135(%629: index):
%630 = arith.cmpi slt, %629, %624 : index
%631 = arith.cmpi sgt, %629, %624 : index
%632 = arith.select %628, %630, %631 : i1
cf.cond_br %632, ^bb136(%629 : index), ^bb137(%629 : index)
^bb136(%633: index):
%634 = llvm.load %616 : !llvm.ptr -> i64
%636 = llvm.mlir.addressof @g_primes : !llvm.ptr
%637 = llvm.load %636 : !llvm.ptr -> !llvm.ptr
%638 = arith.index_cast %633 : index to i64
%639 = llvm.getelementptr %637[%638] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%635 = llvm.load %639 : !llvm.ptr -> i64
%640 = arith.divsi %409, %635 : i64
%641 = arith.cmpi sle, %634, %640 : i64
cf.cond_br %641, ^bb138, ^bb139
^bb138:
%642 = llvm.load %616 : !llvm.ptr -> i64
%644 = llvm.mlir.addressof @g_primes : !llvm.ptr
%645 = llvm.load %644 : !llvm.ptr -> !llvm.ptr
%646 = arith.index_cast %633 : index to i64
%647 = llvm.getelementptr %645[%646] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%643 = llvm.load %647 : !llvm.ptr -> i64
%648 = arith.muli %642, %643 : i64
llvm.store %648, %616 : i64, !llvm.ptr
%649 = llvm.load %620 : !llvm.ptr -> i64
%650 = arith.constant 1 : i32
%652 = arith.extsi %650 : i32 to i64
%651 = arith.addi %649, %652 : i64
llvm.store %651, %620 : i64, !llvm.ptr
cf.br ^bb140
^bb139:
cf.br ^bb137(%633 : index)
^bb140:
%653 = arith.addi %633, %625 : index
cf.br ^bb135(%653 : index)
^bb137(%654: index):
%655 = arith.constant 5000000 : i32
%656 = arith.extsi %655 : i32 to i64
%657 = arith.cmpi sgt, %656, %409 : i64
cf.cond_br %657, ^bb141, ^bb142
^bb141:
%658 = llvm.load %620 : !llvm.ptr -> i64
llvm.store %658, %620 : i64, !llvm.ptr
cf.br ^bb143
^bb142:
cf.br ^bb143
^bb143:
%660 = arith.constant 1 : i32
%662 = arith.extsi %660 : i32 to i64
%661 = arith.addi %656, %662 : i64
%663 = arith.constant 1 : i32
%664 = arith.extsi %663 : i32 to i64
%659 = func.call @calloc(%661, %664) : (i64, i64) -> !llvm.ptr
%666 = arith.constant 1 : i32
%668 = arith.extsi %666 : i32 to i64
%667 = arith.addi %656, %668 : i64
%669 = arith.constant 1 : i32
%670 = arith.extsi %669 : i32 to i64
%665 = func.call @calloc(%667, %670) : (i64, i64) -> !llvm.ptr
%672 = arith.constant 1 : i32
%674 = arith.extsi %672 : i32 to i64
%673 = arith.addi %656, %674 : i64
%675 = arith.constant 1 : i32
%676 = arith.extsi %675 : i32 to i64
%671 = func.call @calloc(%673, %676) : (i64, i64) -> !llvm.ptr
%678 = arith.constant 10 : i32
%680 = arith.extsi %678 : i32 to i64
%679 = arith.divsi %656, %680 : i64
%681 = arith.constant 100 : i32
%683 = arith.extsi %681 : i32 to i64
%682 = arith.addi %679, %683 : i64
%684 = arith.constant 8 : i32
%685 = arith.extsi %684 : i32 to i64
%677 = func.call @calloc(%682, %685) : (i64, i64) -> !llvm.ptr
%686 = arith.constant 0 : i32
%687 = arith.extsi %686 : i32 to i64
%688 = llvm.mlir.constant(1 : i64) : i64
%689 = llvm.alloca %688 x i64 : (i64) -> !llvm.ptr
llvm.store %687, %689 : i64, !llvm.ptr
%690 = arith.constant 1 : i32
%691 = arith.constant 1 : i32
%692 = arith.trunci %690 : i32 to i8
%693 = arith.extsi %691 : i32 to i64
%694 = llvm.getelementptr %659[%693] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %692, %694 : i8, !llvm.ptr
%695 = arith.constant 2 : i32
%696 = arith.constant 1 : i32
%698 = arith.extsi %696 : i32 to i64
%697 = arith.addi %656, %698 : i64
%699 = arith.index_cast %695 : i32 to index
%700 = arith.index_cast %697 : i32 to index
%702 = arith.constant 1 : index
%703 = arith.constant -1 : index
%704 = arith.cmpi sle, %699, %700 : index
%701 = arith.select %704, %702, %703 : index
cf.br ^bb144(%699 : index)
^bb144(%705: index):
%706 = arith.cmpi slt, %705, %700 : index
%707 = arith.cmpi sgt, %705, %700 : index
%708 = arith.select %704, %706, %707 : i1
cf.cond_br %708, ^bb145(%705 : index), ^bb146(%705 : index)
^bb145(%709: index):
%711 = arith.index_cast %709 : index to i64
%712 = llvm.getelementptr %671[%711] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%710 = llvm.load %712 : !llvm.ptr -> i8
%713 = arith.constant 0 : i32
%715 = arith.extsi %710 : i8 to i32
%714 = arith.cmpi eq, %715, %713 : i32
cf.cond_br %714, ^bb147, ^bb148
^bb147:
%716 = llvm.load %689 : !llvm.ptr -> i64
%717 = arith.index_cast %709 : index to i64
%718 = llvm.getelementptr %677[%716] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %717, %718 : i64, !llvm.ptr
%719 = llvm.load %689 : !llvm.ptr -> i64
%720 = arith.constant 1 : i32
%722 = arith.extsi %720 : i32 to i64
%721 = arith.addi %719, %722 : i64
llvm.store %721, %689 : i64, !llvm.ptr
%723 = arith.constant 1 : i32
%725 = arith.constant 0 : i32
%724 = arith.subi %725, %723 : i32
%726 = arith.trunci %724 : i32 to i8
%727 = arith.index_cast %709 : index to i64
%728 = llvm.getelementptr %659[%727] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %726, %728 : i8, !llvm.ptr
%729 = arith.constant 1 : i32
%730 = arith.trunci %729 : i32 to i8
%731 = arith.index_cast %709 : index to i64
%732 = llvm.getelementptr %665[%731] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %730, %732 : i8, !llvm.ptr
cf.br ^bb149
^bb148:
cf.br ^bb149
^bb149:
%733 = arith.constant 0 : i32
%734 = llvm.load %689 : !llvm.ptr -> i64
%735 = arith.index_cast %733 : i32 to index
%736 = arith.index_cast %734 : i32 to index
%738 = arith.constant 1 : index
%739 = arith.constant -1 : index
%740 = arith.cmpi sle, %735, %736 : index
%737 = arith.select %740, %738, %739 : index
cf.br ^bb150(%735 : index)
^bb150(%741: index):
%742 = arith.cmpi slt, %741, %736 : index
%743 = arith.cmpi sgt, %741, %736 : index
%744 = arith.select %740, %742, %743 : i1
cf.cond_br %744, ^bb151(%741 : index), ^bb152(%741 : index)
^bb151(%745: index):
%747 = arith.index_cast %745 : index to i64
%748 = llvm.getelementptr %677[%747] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%746 = llvm.load %748 : !llvm.ptr -> i64
%750 = arith.index_cast %709 : index to i32
%751 = arith.trunci %746 : i64 to i32
%749 = arith.muli %750, %751 : i32
%752 = arith.extsi %749 : i32 to i64
%753 = arith.cmpi sgt, %752, %656 : i64
cf.cond_br %753, ^bb153, ^bb154
^bb153:
cf.br ^bb152(%745 : index)
^bb154:
cf.br ^bb155
^bb155:
%754 = arith.constant 1 : i32
%755 = arith.trunci %754 : i32 to i8
%756 = llvm.getelementptr %671[%752] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %755, %756 : i8, !llvm.ptr
%758 = arith.index_cast %709 : index to i32
%759 = arith.trunci %746 : i64 to i32
%757 = arith.remsi %758, %759 : i32
%760 = arith.constant 0 : i32
%761 = arith.cmpi eq, %757, %760 : i32
cf.cond_br %761, ^bb156, ^bb157
^bb156:
%762 = arith.constant 0 : i32
%763 = arith.trunci %762 : i32 to i8
%764 = llvm.getelementptr %659[%752] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %763, %764 : i8, !llvm.ptr
%766 = arith.index_cast %709 : index to i64
%767 = llvm.getelementptr %665[%766] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%765 = llvm.load %767 : !llvm.ptr -> i8
%768 = llvm.getelementptr %665[%752] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %765, %768 : i8, !llvm.ptr
cf.br ^bb152(%745 : index)
^bb157:
%770 = arith.index_cast %709 : index to i64
%771 = llvm.getelementptr %659[%770] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%769 = llvm.load %771 : !llvm.ptr -> i8
%773 = arith.constant 0 : i8
%772 = arith.subi %773, %769 : i8
%774 = llvm.getelementptr %659[%752] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %772, %774 : i8, !llvm.ptr
%776 = arith.index_cast %709 : index to i64
%777 = llvm.getelementptr %665[%776] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%775 = llvm.load %777 : !llvm.ptr -> i8
%778 = arith.constant 1 : i32
%780 = arith.extsi %775 : i8 to i32
%779 = arith.addi %780, %778 : i32
%781 = arith.trunci %779 : i32 to i8
%782 = llvm.getelementptr %665[%752] : (!llvm.ptr, i64) -> !llvm.ptr, i8
llvm.store %781, %782 : i8, !llvm.ptr
cf.br ^bb158
^bb158:
%783 = arith.addi %745, %737 : index
cf.br ^bb150(%783 : index)
^bb152(%784: index):
%785 = arith.addi %709, %701 : index
cf.br ^bb144(%785 : index)
^bb146(%786: index):
func.call @free(%671) : (!llvm.ptr) -> ()
func.call @free(%677) : (!llvm.ptr) -> ()
%790 = llvm.load %620 : !llvm.ptr -> i64
%791 = arith.constant 1 : i32
%793 = arith.extsi %791 : i32 to i64
%792 = arith.addi %790, %793 : i64
%794 = arith.constant 8 : i32
%795 = arith.extsi %794 : i32 to i64
%789 = func.call @calloc(%792, %795) : (i64, i64) -> !llvm.ptr
%796 = arith.constant 1 : i32
%797 = arith.constant 1 : i32
%799 = arith.extsi %797 : i32 to i64
%798 = arith.addi %656, %799 : i64
%800 = arith.index_cast %796 : i32 to index
%801 = arith.index_cast %798 : i32 to index
%803 = arith.constant 1 : index
%804 = arith.constant -1 : index
%805 = arith.cmpi sle, %800, %801 : index
%802 = arith.select %805, %803, %804 : index
cf.br ^bb159(%800 : index)
^bb159(%806: index):
%807 = arith.cmpi slt, %806, %801 : index
%808 = arith.cmpi sgt, %806, %801 : index
%809 = arith.select %805, %807, %808 : i1
cf.cond_br %809, ^bb160(%806 : index), ^bb161(%806 : index)
^bb160(%810: index):
%812 = arith.index_cast %810 : index to i64
%813 = llvm.getelementptr %659[%812] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%811 = llvm.load %813 : !llvm.ptr -> i8
%814 = arith.constant 0 : i32
%816 = arith.extsi %811 : i8 to i32
%815 = arith.cmpi ne, %816, %814 : i32
cf.cond_br %815, ^bb162, ^bb163
^bb162:
%818 = arith.index_cast %810 : index to i64
%819 = llvm.getelementptr %665[%818] : (!llvm.ptr, i64) -> !llvm.ptr, i8
%817 = llvm.load %819 : !llvm.ptr -> i8
%820 = arith.extsi %817 : i8 to i64
%821 = llvm.load %620 : !llvm.ptr -> i64
%822 = arith.cmpi sle, %820, %821 : i64
cf.cond_br %822, ^bb165, ^bb166
^bb165:
%824 = llvm.getelementptr %789[%820] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%823 = llvm.load %824 : !llvm.ptr -> i64
%825 = arith.index_cast %810 : index to i64
%826 = arith.index_cast %810 : index to i64
%827 = arith.muli %825, %826 : i64
%828 = arith.divsi %408, %827 : i64
%829 = arith.addi %823, %828 : i64
%830 = llvm.getelementptr %789[%820] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %829, %830 : i64, !llvm.ptr
cf.br ^bb167
^bb166:
cf.br ^bb167
^bb167:
cf.br ^bb164
^bb163:
cf.br ^bb164
^bb164:
%831 = arith.addi %810, %802 : index
cf.br ^bb159(%831 : index)
^bb161(%832: index):
func.call @free(%659) : (!llvm.ptr) -> ()
func.call @free(%665) : (!llvm.ptr) -> ()
%835 = arith.constant 1 : i32
%837 = arith.extsi %835 : i32 to i64
%836 = arith.addi %656, %837 : i64
%838 = arith.constant 1 : i32
%840 = arith.extsi %838 : i32 to i64
%839 = arith.addi %656, %840 : i64
%841 = arith.muli %836, %839 : i64
%842 = arith.divsi %408, %841 : i64
%843 = arith.constant 1 : i32
%844 = arith.constant 1 : i32
%846 = arith.extsi %844 : i32 to i64
%845 = arith.addi %842, %846 : i64
%847 = arith.index_cast %843 : i32 to index
%848 = arith.index_cast %845 : i32 to index
%850 = arith.constant 1 : index
%851 = arith.constant -1 : index
%852 = arith.cmpi sle, %847, %848 : index
%849 = arith.select %852, %850, %851 : index
cf.br ^bb168(%847 : index)
^bb168(%853: index):
%854 = arith.cmpi slt, %853, %848 : index
%855 = arith.cmpi sgt, %853, %848 : index
%856 = arith.select %852, %854, %855 : i1
cf.cond_br %856, ^bb169(%853 : index), ^bb170(%853 : index)
^bb169(%857: index):
%860 = arith.trunci %408 : i64 to i32
%861 = arith.index_cast %857 : index to i32
%859 = arith.divsi %860, %861 : i32
%862 = arith.extsi %859 : i32 to i64
%858 = func.call @isqrt(%862) : (i64) -> i64
%864 = arith.constant 1 : i32
%866 = arith.index_cast %857 : index to i32
%865 = arith.addi %866, %864 : i32
%868 = arith.extsi %865 : i32 to i64
%867 = arith.divsi %408, %868 : i64
%863 = func.call @isqrt(%867) : (i64) -> i64
%869 = arith.cmpi sgt, %656, %863 : i64
%870 = scf.if %869 -> (i64) {
scf.yield %656 : i64
} else {
scf.yield %863 : i64
}
%871 = arith.cmpi sle, %858, %870 : i64
cf.cond_br %871, ^bb171, ^bb172
^bb171:
%872 = arith.addi %857, %849 : index
cf.br ^bb168(%872 : index)
^bb172:
cf.br ^bb173
^bb173:
%873 = arith.constant 0 : i32
%874 = llvm.load %620 : !llvm.ptr -> i64
%875 = arith.constant 1 : i32
%877 = arith.extsi %875 : i32 to i64
%876 = arith.addi %874, %877 : i64
%878 = arith.index_cast %873 : i32 to index
%879 = arith.index_cast %876 : i32 to index
%881 = arith.constant 1 : index
%882 = arith.constant -1 : index
%883 = arith.cmpi sle, %878, %879 : index
%880 = arith.select %883, %881, %882 : index
cf.br ^bb174(%878 : index)
^bb174(%884: index):
%885 = arith.cmpi slt, %884, %879 : index
%886 = arith.cmpi sgt, %884, %879 : index
%887 = arith.select %883, %885, %886 : i1
cf.cond_br %887, ^bb175(%884 : index), ^bb176(%884 : index)
^bb175(%888: index):
%890 = arith.index_cast %888 : index to i64
%889 = func.call @prefix_count(%858, %890) : (i64, i64) -> i64
%892 = arith.index_cast %888 : index to i64
%891 = func.call @prefix_count(%870, %892) : (i64, i64) -> i64
%893 = arith.subi %889, %891 : i64
%894 = arith.constant 0 : i32
%896 = arith.extsi %894 : i32 to i64
%895 = arith.cmpi ne, %893, %896 : i64
cf.cond_br %895, ^bb177, ^bb178
^bb177:
%898 = arith.index_cast %888 : index to i64
%899 = llvm.getelementptr %789[%898] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%897 = llvm.load %899 : !llvm.ptr -> i64
%901 = arith.index_cast %857 : index to i32
%902 = arith.trunci %893 : i64 to i32
%900 = arith.muli %901, %902 : i32
%904 = arith.extsi %900 : i32 to i64
%903 = arith.addi %897, %904 : i64
%905 = arith.index_cast %888 : index to i64
%906 = llvm.getelementptr %789[%905] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %903, %906 : i64, !llvm.ptr
cf.br ^bb179
^bb178:
cf.br ^bb179
^bb179:
%907 = arith.addi %888, %880 : index
cf.br ^bb174(%907 : index)
^bb176(%908: index):
%909 = arith.addi %857, %849 : index
cf.br ^bb168(%909 : index)
^bb170(%910: index):
%912 = llvm.load %620 : !llvm.ptr -> i64
%913 = arith.constant 1 : i32
%915 = arith.extsi %913 : i32 to i64
%914 = arith.addi %912, %915 : i64
%916 = llvm.load %620 : !llvm.ptr -> i64
%917 = arith.constant 1 : i32
%919 = arith.extsi %917 : i32 to i64
%918 = arith.addi %916, %919 : i64
%920 = arith.muli %914, %918 : i64
%921 = arith.constant 8 : i32
%922 = arith.extsi %921 : i32 to i64
%911 = func.call @calloc(%920, %922) : (i64, i64) -> !llvm.ptr
%923 = arith.constant 0 : i32
%924 = llvm.load %620 : !llvm.ptr -> i64
%925 = arith.constant 1 : i32
%927 = arith.extsi %925 : i32 to i64
%926 = arith.addi %924, %927 : i64
%928 = arith.index_cast %923 : i32 to index
%929 = arith.index_cast %926 : i32 to index
%931 = arith.constant 1 : index
%932 = arith.constant -1 : index
%933 = arith.cmpi sle, %928, %929 : index
%930 = arith.select %933, %931, %932 : index
cf.br ^bb180(%928 : index)
^bb180(%934: index):
%935 = arith.cmpi slt, %934, %929 : index
%936 = arith.cmpi sgt, %934, %929 : index
%937 = arith.select %933, %935, %936 : i1
cf.cond_br %937, ^bb181(%934 : index), ^bb182(%934 : index)
^bb181(%938: index):
%939 = arith.constant 1 : i32
%940 = llvm.load %620 : !llvm.ptr -> i64
%941 = arith.constant 1 : i32
%943 = arith.extsi %941 : i32 to i64
%942 = arith.addi %940, %943 : i64
%945 = arith.index_cast %938 : index to i32
%946 = arith.trunci %942 : i64 to i32
%944 = arith.muli %945, %946 : i32
%947 = arith.constant 0 : i32
%948 = arith.addi %944, %947 : i32
%949 = arith.extsi %939 : i32 to i64
%950 = arith.extsi %948 : i32 to i64
%951 = llvm.getelementptr %911[%950] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %949, %951 : i64, !llvm.ptr
%952 = arith.constant 1 : i32
%953 = llvm.load %620 : !llvm.ptr -> i64
%954 = arith.constant 1 : i32
%956 = arith.extsi %954 : i32 to i64
%955 = arith.addi %953, %956 : i64
%958 = arith.index_cast %938 : index to i32
%959 = arith.trunci %955 : i64 to i32
%957 = arith.muli %958, %959 : i32
%961 = arith.index_cast %938 : index to i32
%960 = arith.addi %957, %961 : i32
%962 = arith.extsi %952 : i32 to i64
%963 = arith.extsi %960 : i32 to i64
%964 = llvm.getelementptr %911[%963] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %962, %964 : i64, !llvm.ptr
%965 = arith.constant 1 : i32
%966 = arith.index_cast %965 : i32 to index
%967 = arith.index_cast %938 : i32 to index
%969 = arith.constant 1 : index
%970 = arith.constant -1 : index
%971 = arith.cmpi sle, %966, %967 : index
%968 = arith.select %971, %969, %970 : index
cf.br ^bb183(%966 : index)
^bb183(%972: index):
%973 = arith.cmpi slt, %972, %967 : index
%974 = arith.cmpi sgt, %972, %967 : index
%975 = arith.select %971, %973, %974 : i1
cf.cond_br %975, ^bb184(%972 : index), ^bb185(%972 : index)
^bb184(%976: index):
%978 = arith.constant 1 : i32
%980 = arith.index_cast %938 : index to i32
%979 = arith.subi %980, %978 : i32
%981 = llvm.load %620 : !llvm.ptr -> i64
%982 = arith.constant 1 : i32
%984 = arith.extsi %982 : i32 to i64
%983 = arith.addi %981, %984 : i64
%986 = arith.extsi %979 : i32 to i64
%985 = arith.muli %986, %983 : i64
%988 = arith.trunci %985 : i64 to i32
%989 = arith.index_cast %976 : index to i32
%987 = arith.addi %988, %989 : i32
%990 = arith.constant 1 : i32
%991 = arith.subi %987, %990 : i32
%992 = arith.extsi %991 : i32 to i64
%993 = llvm.getelementptr %911[%992] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%977 = llvm.load %993 : !llvm.ptr -> i64
%995 = arith.constant 1 : i32
%997 = arith.index_cast %938 : index to i32
%996 = arith.subi %997, %995 : i32
%998 = llvm.load %620 : !llvm.ptr -> i64
%999 = arith.constant 1 : i32
%1001 = arith.extsi %999 : i32 to i64
%1000 = arith.addi %998, %1001 : i64
%1003 = arith.extsi %996 : i32 to i64
%1002 = arith.muli %1003, %1000 : i64
%1005 = arith.trunci %1002 : i64 to i32
%1006 = arith.index_cast %976 : index to i32
%1004 = arith.addi %1005, %1006 : i32
%1007 = arith.extsi %1004 : i32 to i64
%1008 = llvm.getelementptr %911[%1007] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%994 = llvm.load %1008 : !llvm.ptr -> i64
%1009 = arith.addi %977, %994 : i64
%1010 = llvm.load %620 : !llvm.ptr -> i64
%1011 = arith.constant 1 : i32
%1013 = arith.extsi %1011 : i32 to i64
%1012 = arith.addi %1010, %1013 : i64
%1015 = arith.index_cast %938 : index to i32
%1016 = arith.trunci %1012 : i64 to i32
%1014 = arith.muli %1015, %1016 : i32
%1018 = arith.index_cast %976 : index to i32
%1017 = arith.addi %1014, %1018 : i32
%1019 = arith.extsi %1017 : i32 to i64
%1020 = llvm.getelementptr %911[%1019] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %1009, %1020 : i64, !llvm.ptr
%1021 = arith.addi %976, %968 : index
cf.br ^bb183(%1021 : index)
^bb185(%1022: index):
%1023 = arith.addi %938, %930 : index
cf.br ^bb180(%1023 : index)
^bb182(%1024: index):
%1026 = llvm.load %620 : !llvm.ptr -> i64
%1027 = arith.constant 1 : i32
%1029 = arith.extsi %1027 : i32 to i64
%1028 = arith.addi %1026, %1029 : i64
%1030 = arith.constant 8 : i32
%1031 = arith.extsi %1030 : i32 to i64
%1025 = func.call @calloc(%1028, %1031) : (i64, i64) -> !llvm.ptr
%1032 = arith.constant 0 : i32
%1033 = llvm.load %620 : !llvm.ptr -> i64
%1034 = arith.constant 1 : i32
%1036 = arith.extsi %1034 : i32 to i64
%1035 = arith.addi %1033, %1036 : i64
%1037 = arith.index_cast %1032 : i32 to index
%1038 = arith.index_cast %1035 : i32 to index
%1040 = arith.constant 1 : index
%1041 = arith.constant -1 : index
%1042 = arith.cmpi sle, %1037, %1038 : index
%1039 = arith.select %1042, %1040, %1041 : index
cf.br ^bb186(%1037 : index)
^bb186(%1043: index):
%1044 = arith.cmpi slt, %1043, %1038 : index
%1045 = arith.cmpi sgt, %1043, %1038 : index
%1046 = arith.select %1042, %1044, %1045 : i1
cf.cond_br %1046, ^bb187(%1043 : index), ^bb188(%1043 : index)
^bb187(%1047: index):
%1048 = llvm.load %620 : !llvm.ptr -> i64
%1050 = arith.trunci %1048 : i64 to i32
%1051 = arith.index_cast %1047 : index to i32
%1049 = arith.subi %1050, %1051 : i32
%1052 = arith.extsi %1049 : i32 to i64
%1054 = llvm.getelementptr %789[%1052] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%1053 = llvm.load %1054 : !llvm.ptr -> i64
%1055 = llvm.mlir.constant(1 : i64) : i64
%1056 = llvm.alloca %1055 x i64 : (i64) -> !llvm.ptr
llvm.store %1053, %1056 : i64, !llvm.ptr
%1057 = arith.constant 1 : i32
%1059 = arith.extsi %1057 : i32 to i64
%1058 = arith.addi %1052, %1059 : i64
%1060 = llvm.load %620 : !llvm.ptr -> i64
%1061 = arith.constant 1 : i32
%1063 = arith.extsi %1061 : i32 to i64
%1062 = arith.addi %1060, %1063 : i64
%1064 = arith.index_cast %1058 : i32 to index
%1065 = arith.index_cast %1062 : i32 to index
%1067 = arith.constant 1 : index
%1068 = arith.constant -1 : index
%1069 = arith.cmpi sle, %1064, %1065 : index
%1066 = arith.select %1069, %1067, %1068 : index
cf.br ^bb189(%1064 : index)
^bb189(%1070: index):
%1071 = arith.cmpi slt, %1070, %1065 : index
%1072 = arith.cmpi sgt, %1070, %1065 : index
%1073 = arith.select %1069, %1071, %1072 : i1
cf.cond_br %1073, ^bb190(%1070 : index), ^bb191(%1070 : index)
^bb190(%1074: index):
%1075 = llvm.load %1056 : !llvm.ptr -> i64
%1077 = arith.index_cast %1074 : index to i64
%1078 = llvm.getelementptr %1025[%1077] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%1076 = llvm.load %1078 : !llvm.ptr -> i64
%1080 = llvm.load %620 : !llvm.ptr -> i64
%1081 = arith.constant 1 : i32
%1083 = arith.extsi %1081 : i32 to i64
%1082 = arith.addi %1080, %1083 : i64
%1085 = arith.index_cast %1074 : index to i32
%1086 = arith.trunci %1082 : i64 to i32
%1084 = arith.muli %1085, %1086 : i32
%1088 = arith.extsi %1084 : i32 to i64
%1087 = arith.addi %1088, %1052 : i64
%1089 = llvm.getelementptr %911[%1087] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%1079 = llvm.load %1089 : !llvm.ptr -> i64
%1090 = arith.muli %1076, %1079 : i64
%1091 = arith.subi %1075, %1090 : i64
llvm.store %1091, %1056 : i64, !llvm.ptr
%1092 = arith.addi %1074, %1066 : index
cf.br ^bb189(%1092 : index)
^bb191(%1093: index):
%1094 = llvm.load %1056 : !llvm.ptr -> i64
%1095 = llvm.getelementptr %1025[%1052] : (!llvm.ptr, i64) -> !llvm.ptr, i64
llvm.store %1094, %1095 : i64, !llvm.ptr
%1096 = arith.addi %1047, %1039 : index
cf.br ^bb186(%1096 : index)
^bb188(%1097: index):
%1098 = arith.constant 1 : i32
%1099 = arith.extsi %1098 : i32 to i64
%1100 = llvm.mlir.constant(1 : i64) : i64
%1101 = llvm.alloca %1100 x i64 : (i64) -> !llvm.ptr
llvm.store %1099, %1101 : i64, !llvm.ptr
%1102 = arith.constant 0 : i32
%1103 = llvm.load %620 : !llvm.ptr -> i64
%1104 = arith.constant 1 : i32
%1106 = arith.extsi %1104 : i32 to i64
%1105 = arith.addi %1103, %1106 : i64
%1107 = arith.index_cast %1102 : i32 to index
%1108 = arith.index_cast %1105 : i32 to index
%1110 = arith.constant 1 : index
%1111 = arith.constant -1 : index
%1112 = arith.cmpi sle, %1107, %1108 : index
%1109 = arith.select %1112, %1110, %1111 : index
cf.br ^bb192(%1107 : index)
^bb192(%1113: index):
%1114 = arith.cmpi slt, %1113, %1108 : index
%1115 = arith.cmpi sgt, %1113, %1108 : index
%1116 = arith.select %1112, %1114, %1115 : i1
cf.cond_br %1116, ^bb193(%1113 : index), ^bb194(%1113 : index)
^bb193(%1117: index):
%1119 = arith.index_cast %1117 : index to i64
%1120 = llvm.getelementptr %1025[%1119] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%1118 = llvm.load %1120 : !llvm.ptr -> i64
%1121 = arith.constant 0 : i32
%1123 = arith.extsi %1121 : i32 to i64
%1122 = arith.cmpi ne, %1118, %1123 : i64
cf.cond_br %1122, ^bb195, ^bb196
^bb195:
%1125 = arith.index_cast %1117 : index to i64
%1126 = llvm.getelementptr %1025[%1125] : (!llvm.ptr, i64) -> !llvm.ptr, i64
%1124 = llvm.load %1126 : !llvm.ptr -> i64
%1127 = llvm.mlir.addressof @MOD : !llvm.ptr
%1128 = llvm.load %1127 : !llvm.ptr -> i64
%1129 = arith.remsi %1124, %1128 : i64
%1130 = arith.constant 0 : i32
%1132 = arith.extsi %1130 : i32 to i64
%1131 = arith.cmpi slt, %1129, %1132 : i64
%1133 = scf.if %1131 -> (i64) {
%1134 = llvm.mlir.addressof @MOD : !llvm.ptr
%1135 = llvm.load %1134 : !llvm.ptr -> i64
%1136 = arith.addi %1129, %1135 : i64
scf.yield %1136 : i64
} else {
scf.yield %1129 : i64
}
%1137 = llvm.load %1101 : !llvm.ptr -> i64
%1138 = arith.muli %1137, %1133 : i64
%1139 = llvm.mlir.addressof @MOD : !llvm.ptr
%1140 = llvm.load %1139 : !llvm.ptr -> i64
%1141 = arith.remsi %1138, %1140 : i64
llvm.store %1141, %1101 : i64, !llvm.ptr
cf.br ^bb197
^bb196:
cf.br ^bb197
^bb197:
%1142 = arith.addi %1117, %1109 : index
cf.br ^bb192(%1142 : index)
^bb194(%1143: index):
%1144 = llvm.mlir.addressof @str_0 : !llvm.ptr
%1145 = llvm.load %1101 : !llvm.ptr -> i64
%1146 = llvm.call @printf(%1144, %1145) vararg(!llvm.func<i32 (ptr, ...)>) : (!llvm.ptr, i64) -> i32
func.call @free(%1025) : (!llvm.ptr) -> ()
func.call @free(%911) : (!llvm.ptr) -> ()
func.call @free(%789) : (!llvm.ptr) -> ()
%1151 = llvm.mlir.addressof @g_memo_val : !llvm.ptr
%1152 = llvm.load %1151 : !llvm.ptr -> !llvm.ptr
func.call @free(%1152) : (!llvm.ptr) -> ()
%1154 = llvm.mlir.addressof @g_memo_key : !llvm.ptr
%1155 = llvm.load %1154 : !llvm.ptr -> !llvm.ptr
func.call @free(%1155) : (!llvm.ptr) -> ()
%1157 = llvm.mlir.addressof @g_primes : !llvm.ptr
%1158 = llvm.load %1157 : !llvm.ptr -> !llvm.ptr
func.call @free(%1158) : (!llvm.ptr) -> ()
%1159 = arith.constant 0 : i32
func.return %1159 : i32
}
}