import sys import sys import taichi as ti import math import numpy as np import cv2 import os import matplotlib.pyplot as plt import time from matplotlib.pyplot import cm import taichi as tc real = ti.f32 ti.set_default_fp(real) max_steps = 4096 vis_interval = 4 output_vis_interval = 16 steps = 204 assert steps * 2 <= max_steps vis_resolution = 1024 scalar = lambda: ti.var(dt=real) vec = lambda: ti.Vector(2, dt=real) loss = scalar() x = vec() v = vec()
def test_atan2_f64(): ti.set_default_fp(ti.f64) grad_test(lambda x: ti.atan2(0.4, x), lambda x: np.arctan2(0.4, x)) grad_test(lambda y: ti.atan2(y, 0.4), lambda y: np.arctan2(y, 0.4))