Add forwardDType to Cast operator.

include/aidge/operator/Cast.hpp

@@ -98,6 +98,12 @@ public:
      */
     Cast_Op(const Cast_Op& op);
 
+    /**
+     * @brief Forward the data type.
+     * @return True if successful, false otherwise.
+     */
+    bool forwardDType() override final;
+
     /**
      * @brief Clone the operator using its copy constructor.
      * @return A shared pointer to the cloned operator.

src/operator/Cast.cpp

@@ -49,6 +49,11 @@ Cast_Op::Cast_Op(const Cast_Op& op)
     }
 }
 
+bool Aidge::Cast_Op::forwardDType(){
+    mOutputs[0]->setDataType(mAttributes->getAttr<CastAttr::TargetType>());
+    return true;
+}
+
 void Aidge::Cast_Op::setDataType(const DataType& dataType) const {
     if (targetType() != dataType) {
         Log::warn("Cast::setDataType(): Cannot setDataType for cast operator.");