diff --git a/python/paddle/v2/framework/tests/test_nccl_ops.py b/python/paddle/v2/framework/tests/test_nccl_ops.py new file mode 100644 index 0000000000000000000000000000000000000000..128a9ab21a38831e9a7316e3f11d1849d313ddb3 --- /dev/null +++ b/python/paddle/v2/framework/tests/test_nccl_ops.py @@ -0,0 +1,17 @@ +import unittest, os +import numpy as np +import paddle.v2 as paddle +from paddle.v2.framework.op import Operator +import paddle.v2.framework.core as core +from op_test import OpTest, create_op + +gpu_list = os.environ["NV_LIST"] + +if not core.is_compile_gpu() or not gpu_list: + exit(0) + + +class TestNCCLAllReduce(unittest.TestCase): + def __init__(self): + self.op_type = "nnclAllReduce" + self.scope = core.Scope()