Update

[ghstack-poisoned]

Author: GregoryComer

.ci/scripts/wheel/test_windows.py

@@ -53,7 +53,7 @@ def test_model_xnnpack(model: Model, quantize: bool) -> None:
 
     assert len(ref_outputs) == len(et_outputs)
     for i in range(len(ref_outputs)):
-        assert torch.allclose(ref_outputs[i], et_outputs[i], atol=1e-4)
+        torch.testing.assert_close(ref_outputs[i], et_outputs[i], atol=1e-4)
 
 
 def run_tests(model_tests: List[ModelTest]) -> None: