bugfix: fix incorrect parsing of float32 dtype in parse_dtype. (#438)

wly-115 · web-flow · commit 7de824c69623 · 2025-11-27T09:35:50.000+08:00
diff --git a/xllm/core/util/utils.cpp b/xllm/core/util/utils.cpp
@@ -80,7 +80,7 @@ torch::ScalarType parse_dtype(const std::string& dtype_str,
   }
   if ((boost::iequals(dtype_str, "float") ||
        boost::iequals(dtype_str, "float32"))) {
-    return torch::kFloat16;
+    return torch::kFloat;
   }
 
   if (dtype_str.empty() || boost::iequals(dtype_str, "auto")) {

Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,7 @@ torch::ScalarType parse_dtype(const std::string& dtype_str,`
`80`	`80`	`}`
`81`	`81`	`if ((boost::iequals(dtype_str, "float") \|\|`
`82`	`82`	`boost::iequals(dtype_str, "float32"))) {`
`83`		`- return torch::kFloat16;`
	`83`	`+ return torch::kFloat;`
`84`	`84`	`}`
`85`	`85`
`86`	`86`	`if (dtype_str.empty() \|\| boost::iequals(dtype_str, "auto")) {`