DMTPARK / mytetration

Make your own tetration Image!
53 stars 23 forks source link

cuda 버전 추가 / zoom_in_out 기능 추가 #7

Open creeras opened 1 month ago

creeras commented 1 month ago

cuda 버전 추가되었습니다. cuda_zoomin 기능 추가되었습니다. 재실행 없이 화면에서 그대로 변경이 적용됩니다.

sasak2 commented 1 month ago

colab에서 런타임 T4 GPU로 실행해보니 n=500기준 2~3초만에 나오네요 :+1: :+1: 선생님 너무 좋습니다!! 따로 코랩에 이식해서 내면 많은 사람들에게 접근성도 좋고 속도도 빠르게 할 수 있을 것 같아요!!

chitenshi commented 1 month ago

colab에서 돌려봤는데 정말 빠릅니다 compute_tetration_divergence함수에서 반복문 부분을

    for k i tqdm(range(max_iter), desc="Calculating Tetration", ncols=100):
        cp.power(c, z, out=z)
    cp.logical_or(cp.isinf(z), cp.isnan(z), out=divergence_map)
    mask = cp.abs(z) > escape_radius
    divergence_map[mask] = True

이렇게 수정하면 좀 더 빠르게 작동하지 않을까 생각합니다 이 코드를 당장 colab에서 테스트해보고 싶었지만 할당량이 넘어서 미리 실험해보지 못했습니다.