From 941b13260952e2b96d4b5fea00dd19eec7d37597 Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Thu, 22 Aug 2024 16:14:04 +0900 Subject: [PATCH] use `torch.testing.assert_close` Signed-off-by: Masaki Kozuki --- thunder/tests/test_inplace_functionalization.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/thunder/tests/test_inplace_functionalization.py b/thunder/tests/test_inplace_functionalization.py index 9d02f52740..ea60155a7c 100644 --- a/thunder/tests/test_inplace_functionalization.py +++ b/thunder/tests/test_inplace_functionalization.py @@ -580,9 +580,9 @@ def f(x, y, idx, src): jitted = executor.make_callable(f) o = jitted(a, b, idx, src) - assert a.allclose(a_) - assert b.allclose(b_) - assert o.allclose(o_) + torch.testing.assert_close(a, a_) + torch.testing.assert_close(b, b_) + torch.testing.assert_close(o, o_) @instantiate(