test_multigpu.py 221 字节
Newer Older
D
Dong Zhihong 已提交
1 2 3 4 5 6 7 8
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, set_input

gpu_list = "0,1,2,3"