Luo Tao
7 years ago
commit
5dc63398b6
@ -1,6 +1,6 @@
|
||||
if(WITH_PYTHON)
|
||||
cc_library(paddle_pybind SHARED
|
||||
SRCS pybind.cc
|
||||
SRCS pybind.cc protobuf.cc
|
||||
DEPS pybind python backward
|
||||
${GLOB_OP_LIB})
|
||||
endif(WITH_PYTHON)
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,35 @@
|
||||
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License. */
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <Python.h>
|
||||
#include <fstream>
|
||||
#include <vector>
|
||||
#include "paddle/framework/op_registry.h"
|
||||
#include "pybind11/numpy.h"
|
||||
#include "pybind11/pybind11.h"
|
||||
#include "pybind11/stl.h"
|
||||
|
||||
namespace py = pybind11;
|
||||
|
||||
namespace paddle {
|
||||
namespace pybind {
|
||||
|
||||
void BindProgramDesc(py::module& m);
|
||||
void BindBlockDesc(py::module& m);
|
||||
void BindVarDsec(py::module& m);
|
||||
void BindOpDesc(py::module& m);
|
||||
} // namespace pybind
|
||||
} // namespace paddle
|
||||
@ -0,0 +1,131 @@
|
||||
import unittest
|
||||
import paddle.v2.framework.core as core
|
||||
|
||||
|
||||
class TestOpDesc(unittest.TestCase):
|
||||
def test_op_desc(self):
|
||||
prog = core.ProgramDesc.__create_program_desc__()
|
||||
self.assertIsNotNone(prog)
|
||||
block = prog.block(0)
|
||||
self.assertIsNotNone(block)
|
||||
op = block.append_op()
|
||||
self.assertIsNotNone(op)
|
||||
op.set_type("test")
|
||||
self.assertEqual("test", op.type())
|
||||
op.set_input("X", ["a", "b", "c"])
|
||||
self.assertEqual(["a", "b", "c"], op.input("X"))
|
||||
self.assertEqual(["X"], op.input_names())
|
||||
|
||||
op.set_output("Out", ["z"])
|
||||
self.assertEqual(['z'], op.output("Out"))
|
||||
self.assertEqual(["Out"], op.output_names())
|
||||
|
||||
op.set_attr("int_attr", 1)
|
||||
self.assertEqual(1, op.attr("int_attr"))
|
||||
self.assertTrue(op.has_attr("int_attr"))
|
||||
self.assertEqual(core.AttrType.INT, op.attr_type("int_attr"))
|
||||
|
||||
op.set_attr("float_attr", -1.32)
|
||||
self.assertAlmostEqual(-1.32, op.attr("float_attr"), delta=1e-4)
|
||||
self.assertTrue(op.has_attr("float_attr"))
|
||||
|
||||
op.set_attr("bool_attr", False)
|
||||
self.assertFalse(op.attr("bool_attr"))
|
||||
|
||||
op.set_attr("string_attr", "abc")
|
||||
self.assertEqual("abc", op.attr("string_attr"))
|
||||
self.assertTrue(op.has_attr("string_attr"))
|
||||
|
||||
op.set_attr("ints_attr", [1, 2, 3])
|
||||
self.assertEqual([1, 2, 3], op.attr("ints_attr"))
|
||||
|
||||
expected = [1.2, 2.3, 3.4]
|
||||
op.set_attr("floats_attr", expected)
|
||||
for e, a in zip(expected, op.attr("floats_attr")):
|
||||
self.assertAlmostEqual(e, a, delta=1e-4)
|
||||
|
||||
op.set_attr("strings_attr", ["a", "b", "c"])
|
||||
self.assertEqual(["a", "b", "c"], op.attr("strings_attr"))
|
||||
|
||||
op.set_attr("bools_attr", [True, False, True])
|
||||
self.assertEqual([True, False, True], op.attr("bools_attr"))
|
||||
|
||||
self.assertEqual(8, len(op.attr_names()))
|
||||
|
||||
op.set_block_attr("block_attr", prog.block(0))
|
||||
self.assertEqual(0, op.get_block_attr("block_attr"))
|
||||
|
||||
|
||||
class TestProgramDesc(unittest.TestCase):
|
||||
def test_instance(self):
|
||||
program_desc = core.ProgramDesc.__create_program_desc__()
|
||||
self.assertIsNotNone(program_desc)
|
||||
del program_desc
|
||||
program_desc = core.ProgramDesc.instance()
|
||||
self.assertIsNotNone(program_desc)
|
||||
self.assertIsNotNone(program_desc.block(0))
|
||||
del program_desc
|
||||
|
||||
def test_append_block(self):
|
||||
prog_desc = core.ProgramDesc.__create_program_desc__()
|
||||
self.assertIsNotNone(prog_desc)
|
||||
block_root = prog_desc.block(0)
|
||||
self.assertIsNotNone(block_root)
|
||||
self.assertEqual(block_root.id, 0)
|
||||
block1 = prog_desc.append_block(block_root)
|
||||
block2 = prog_desc.append_block(block1)
|
||||
self.assertIsNotNone(block1)
|
||||
self.assertEqual(block1.id, block2.parent)
|
||||
self.assertEqual(block_root.id, block1.parent)
|
||||
block3 = prog_desc.append_block(block_root)
|
||||
self.assertEqual(block3.parent, block_root.id)
|
||||
self.assertEqual(prog_desc.block(1).id, 1)
|
||||
self.assertEqual(4, prog_desc.num_blocks())
|
||||
|
||||
|
||||
class TestVarDesc(unittest.TestCase):
|
||||
def test_shape(self):
|
||||
program_desc = core.ProgramDesc.__create_program_desc__()
|
||||
block = program_desc.block(0)
|
||||
var = block.new_var('my_var')
|
||||
src_shape = [3, 2, 10, 8]
|
||||
var.set_shape(src_shape)
|
||||
res_shape = var.shape()
|
||||
self.assertEqual(src_shape, res_shape)
|
||||
|
||||
def test_data_type(self):
|
||||
program_desc = core.ProgramDesc.__create_program_desc__()
|
||||
block = program_desc.block(0)
|
||||
var = block.new_var('my_var')
|
||||
var.set_data_type(core.DataType.INT32)
|
||||
self.assertEqual(core.DataType.INT32, var.data_type())
|
||||
|
||||
|
||||
class TestBlockDesc(unittest.TestCase):
|
||||
def test_add_var(self):
|
||||
prog = core.ProgramDesc.__create_program_desc__()
|
||||
self.assertIsNotNone(prog)
|
||||
block = prog.block(0)
|
||||
self.assertIsNotNone(block)
|
||||
var1 = block.new_var("var1")
|
||||
var2 = block.new_var("var2")
|
||||
var3 = block.new_var("var3")
|
||||
all_vars = block.all_vars()
|
||||
self.assertEqual(set(all_vars), set([var1, var2, var3]))
|
||||
var2_re = block.var("var2")
|
||||
self.assertEqual(var2_re, var2)
|
||||
|
||||
def test_add_op(self):
|
||||
prog = core.ProgramDesc.__create_program_desc__()
|
||||
self.assertIsNotNone(prog)
|
||||
block = prog.block(0)
|
||||
self.assertIsNotNone(block)
|
||||
op1 = block.append_op()
|
||||
op2 = block.append_op()
|
||||
op0 = block.prepend_op()
|
||||
all_ops = block.all_ops()
|
||||
self.assertEqual(all_ops, [op0, op1, op2])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Loading…
Reference in new issue