failed to optimize python source codes by returning value directly

[SNSerHello]

-------------------OKay's Source Codes---------------------

@ti.func
def taichi_is_prime(n: int) -> bool:
    result = True
    for k in range(2, int(n ** 0.5) + 1):
        if n % k == 0:
            result = False
            break
    return result

@ti.kernel
def taichi_prime_count(n: int) -> int:
    counter = 0
    for k in range(2, n + 1):
        if taichi_is_prime(n):
            counter += 1
    return counter

-----------------Failure's Source Codes---------------------

@ti.func
def taichi_is_prime(n: int) -> bool:
    for k in range(2, int(n ** 0.5) + 1):
        if n % k == 0:
            return False
    return True

@ti.kernel
def taichi_prime_count(n: int) -> int:
    counter = 0
    for k in range(2, n + 1):
        if taichi_is_prime(n):
            counter += 1
    return counter

Why? bug?

[bobcao3]

We don't support multiple return statements yet