Method test_round_gpu

test/python/test_operation.py:3529–3530

(self)

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

    def embedding_helper(self, dev):
        embedding = layer.Embedding(10, 3)