include/aidge/utils/Registrar.hpp

@@ -132,11 +132,13 @@ void declare_registrable(py::module& m, const std::string& class_name){
 #ifdef PYBIND
 #define SET_IMPL_MACRO(T_Op, op, backend_name) \
      \
-        if(Py_IsInitialized()) { \
-            auto obj = py::cast(&(op)); \
-            (op).setImpl(Registrar<T_Op>::create(backend_name)(op)); \
-        } else { \
-            (op).setImpl(Registrar<T_Op>::create(backend_name)(op)); \
+        if (Registrar<T_Op>::exists(backend_name)) { \
+            if(Py_IsInitialized()) { \
+                auto obj = py::cast(&(op)); \
+                (op).setImpl(Registrar<T_Op>::create(backend_name)(op)); \
+            } else { \
+                (op).setImpl(Registrar<T_Op>::create(backend_name)(op)); \
+            } \
         }
 #else
 #define SET_IMPL_MACRO(T_Op, op, backend_name)                          \