提交 51abb6c3 编写于 作者: D Dong Zhihong

add test

上级 4118782b
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()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册