triton.testing.assert_close

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

断言两个输入在一定容差范围内接近。

参数:

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

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

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

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

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