From e17996624873ab76b2dd3a18c1441fbc506f0910 Mon Sep 17 00:00:00 2001 From: Sandeep Kumar Pani Date: Tue, 28 Jul 2020 07:48:47 -0700 Subject: [PATCH] [caffe2][tpx] log to stderr (#42162) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/42162 Test Plan: CI Reviewed By: mrshenli Differential Revision: D22791440 fbshipit-source-id: 14f16cd7a94a57161c5724177b518527f486232d --- test/distributed/test_nccl.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/distributed/test_nccl.py b/test/distributed/test_nccl.py index 1bc3342b1090e..6a56b2625a0c0 100644 --- a/test/distributed/test_nccl.py +++ b/test/distributed/test_nccl.py @@ -18,7 +18,8 @@ nGPUs = torch.cuda.device_count() if not TEST_CUDA: - print('CUDA not available, skipping tests') + import sys + print('CUDA not available, skipping tests', file=sys.stderr) TestCase = object # noqa: F811