diff --git a/src/py/migraphx_py.cpp b/src/py/migraphx_py.cpp
index 5609d5ebe7b..ee830bb4773 100644
--- a/src/py/migraphx_py.cpp
+++ b/src/py/migraphx_py.cpp
@@ -25,6 +25,7 @@
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 #include <pybind11/numpy.h>
+#include <pybind11/operators.h>
 #include <migraphx/program.hpp>
 #include <migraphx/instruction_ref.hpp>
 #include <migraphx/operation.hpp>
@@ -419,9 +420,9 @@ MIGRAPHX_PYBIND11_MODULE(migraphx, m)
         .def("op", [](migraphx::instruction_ref i) { return i->get_operator(); })
         .def("inputs", [](migraphx::instruction_ref i) { return i->inputs(); })
         .def("name", [](migraphx::instruction_ref i) { return i->name(); })
-        .def("__hash__", std::hash<migraphx::instruction_ref>{})
-        .def("__eq__", std::equal_to<migraphx::instruction_ref>{})
-        .def("__eq__", std::equal_to<py::object>{});
+        .def(py::hash(py::self))
+        .def(py::self == py::self)
+        .def(py::self != py::self);
 
     py::class_<migraphx::module, std::unique_ptr<migraphx::module, py::nodelete>>(m, "module")
         .def("print", [](const migraphx::module& mm) { std::cout << mm << std::endl; })