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