Method test_max_gpu

test/python/test_operation.py:1562–1563

(self)

    @unittest.skipIf(not singa_wrap.USE_CUDA, 'CUDA is not enabled')
    def test_max_gpu(self):
        self._max_helper(gpu_dev)

    def _max_3inputs_helper(self, dev):
        data_0 = np.array([3, 2, 1]).astype(np.float32)