未验证 提交 8788513b 编写于 作者: H Hui Zhang 提交者: GitHub

[jit] rm useless property pybind (#44962)

* rm useless pybind

* rm useless ut
上级 a52357fe
...@@ -410,10 +410,6 @@ void BindJitProperty(pybind11::module *m) { ...@@ -410,10 +410,6 @@ void BindJitProperty(pybind11::module *m) {
[](jit::Property &self) { new (&self) jit::Property(); }, [](jit::Property &self) { new (&self) jit::Property(); },
pybind11::return_value_policy::reference) pybind11::return_value_policy::reference)
.def("size", &jit::Property::Size) .def("size", &jit::Property::Size)
.def("set_float",
py::overload_cast<const float &>(&jit::Property::SetFloat),
"set float",
py::arg("val"))
.def("set_float", .def("set_float",
py::overload_cast<const std::string &, const float &>( py::overload_cast<const std::string &, const float &>(
&jit::Property::SetFloat), &jit::Property::SetFloat),
...@@ -425,53 +421,30 @@ void BindJitProperty(pybind11::module *m) { ...@@ -425,53 +421,30 @@ void BindJitProperty(pybind11::module *m) {
.def("get_float", .def("get_float",
py::overload_cast<const std::string &>(&jit::Property::GetFloat, py::overload_cast<const std::string &>(&jit::Property::GetFloat,
py::const_)) py::const_))
.def("set_floats",
py::overload_cast<const std::vector<float> &>(
&jit::Property::SetFloats),
"set list of float",
py::arg("vals"))
.def("set_floats", .def("set_floats",
py::overload_cast<const std::string &, const std::vector<float> &>( py::overload_cast<const std::string &, const std::vector<float> &>(
&jit::Property::SetFloats), &jit::Property::SetFloats),
"set list of float", "set list of float",
py::arg("name"), py::arg("name"),
py::arg("val")) py::arg("val"))
.def("set_int",
py::overload_cast<const int64_t &>(&jit::Property::SetInt64),
"set int",
py::arg("val"))
.def("set_int", .def("set_int",
py::overload_cast<const std::string &, const int64_t &>( py::overload_cast<const std::string &, const int64_t &>(
&jit::Property::SetInt64), &jit::Property::SetInt64),
"set int", "set int",
py::arg("name"), py::arg("name"),
py::arg("val")) py::arg("val"))
.def("set_ints",
py::overload_cast<const std::vector<int64_t> &>(
&jit::Property::SetInt64s),
"set list of int",
py::arg("vals"))
.def("set_ints", .def("set_ints",
py::overload_cast<const std::string &, const std::vector<int64_t> &>( py::overload_cast<const std::string &, const std::vector<int64_t> &>(
&jit::Property::SetInt64s), &jit::Property::SetInt64s),
"set list of int", "set list of int",
py::arg("name"), py::arg("name"),
py::arg("val")) py::arg("val"))
.def("set_string",
py::overload_cast<const std::string &>(&jit::Property::SetString),
"set string",
py::arg("val"))
.def("set_string", .def("set_string",
py::overload_cast<const std::string &, const std::string &>( py::overload_cast<const std::string &, const std::string &>(
&jit::Property::SetString), &jit::Property::SetString),
"set string", "set string",
py::arg("name"), py::arg("name"),
py::arg("val")) py::arg("val"))
.def("set_strings",
py::overload_cast<const std::vector<std::string> &>(
&jit::Property::SetStrings),
"set list of string",
py::arg("vals"))
.def("set_strings", .def("set_strings",
py::overload_cast<const std::string &, py::overload_cast<const std::string &,
const std::vector<std::string> &>( const std::vector<std::string> &>(
......
...@@ -44,29 +44,16 @@ class TestPropertySave(unittest.TestCase): ...@@ -44,29 +44,16 @@ class TestPropertySave(unittest.TestCase):
with self.assertRaises(ValueError): with self.assertRaises(ValueError):
self.a.get_float(1) self.a.get_float(1)
def test_set_float_wo_name(self):
"""test save without name
"""
a = paddle.framework.core.Property()
a.set_float(10.0)
self.assertEqual(a.get_float(0), 10.0)
def test_set(self): def test_set(self):
"""test propety set. """test propety set.
""" """
try: try:
a = paddle.framework.core.Property() a = paddle.framework.core.Property()
a.set_float(10.0)
a.set_float('float', 10.0) a.set_float('float', 10.0)
a.set_floats([5.0, 4.0, 3.0])
a.set_floats('floats', [5.0, 4.0, 3.0]) a.set_floats('floats', [5.0, 4.0, 3.0])
a.set_int(5)
a.set_int('int', 5) a.set_int('int', 5)
a.set_ints([1, 2, 3])
a.set_ints('ints', [1, 2, 3]) a.set_ints('ints', [1, 2, 3])
a.set_string("hello")
a.set_string("str", "hello") a.set_string("str", "hello")
a.set_strings(["1", "2", "3"])
a.set_strings('strs', ["1", "2", "3"]) a.set_strings('strs', ["1", "2", "3"])
except Exception as e: except Exception as e:
self.assertEqual(False, True) self.assertEqual(False, True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册