def rsqrt(poly_sch, fuzz_shape=None): test_ms_rsqrt((32, 1024, 1024), 'float32', poly_sch=poly_sch) test_ms_rsqrt((32, 1024, 1024), 'float16', poly_sch=poly_sch)
def test_rsqrt(): test_ms_rsqrt((32, 1024, 1024), 'float32', poly_sch=True) test_ms_rsqrt((32, 1024, 1024), 'float16', poly_sch=True) return True