triton.testing.assert_close

triton.testing.assert_close(x, y, atol=None, rtol=None, err_msg='')

断言两个输入在某个容差范围内是接近的。

Parameters:

• x (scala, list, numpy.ndarray, or torch.Tensor) – 第一个输入。

• y (scala, list, numpy.ndarray, or torch.Tensor) – 第二个输入。

• atol (float, optional) – 绝对容差。默认值为1e-2。

• rtol (float, optional) – 相对容差。默认值为0。

• err_msg (str) - 断言失败时使用的错误信息。