support model convert from fp32 to fp16

xiaoyewww · xiaoyewww · commit 710c56a7a385 · 2024-06-20T07:13:09.000Z
diff --git a/paddle2onnx/__init__.py b/paddle2onnx/__init__.py
@@ -11,7 +11,6 @@
 # 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.
-
 from .version import version
 
 __version__ = version
diff --git a/paddle2onnx/mapper/mapper.h b/paddle2onnx/mapper/mapper.h
@@ -22,7 +22,7 @@
 
 namespace paddle2onnx {
 
-  static const std::unordered_set<int32_t> kNoNeedCastTypes{P2ODataType::INT8, P2ODataType::FP16, P2ODataType::FP32}; // 0: float32, 3: int8, 6: float16
+  static const std::unordered_set<int32_t> kNoNeedCastTypes{P2ODataType::INT8, P2ODataType::FP16, P2ODataType::FP32};
 
   class Mapper
   {

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@`
`22`	`22`
`23`	`23`	`namespace paddle2onnx {`
`24`	`24`
`25`		`- static const std::unordered_set<int32_t> kNoNeedCastTypes{P2ODataType::INT8, P2ODataType::FP16, P2ODataType::FP32}; // 0: float32, 3: int8, 6: float16`
	`25`	`+ static const std::unordered_set<int32_t> kNoNeedCastTypes{P2ODataType::INT8, P2ODataType::FP16, P2ODataType::FP32};`
`26`	`26`
`27`	`27`	`class Mapper`
`28`	`28`	`{`