parent
4118782b31
commit
51abb6c323
@ -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()
|
Loading…
Reference in new issue