triton.testing.assert_close

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

Asserts that two inputs are close within a certain tolerance.

参数:

• x (scala, list, numpy.ndarray, or torch.Tensor) -- The first input.

• y (scala, list, numpy.ndarray, or torch.Tensor) -- The second input.

• atol (float, optional) -- The absolute tolerance. Default value is 1e-2.

• rtol (float, optional) -- The relative tolerance. Default value is 0.

• err_msg (str) -- The error message to use if the assertion fails.