gpu update cast:fp16->int32

f456f5fc · VectorSL · 8f9af74f · f456f5fc
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

python/akg/ms/gpu/cast.py python/akg/ms/gpu/cast.py +2 -0

未找到文件。
--- a/python/akg/ms/gpu/cast.py
+++ b/python/akg/ms/gpu/cast.py
@@ -22,4 +22,6 @@ def Cast(x, dst_type):
    """cast."""
    if x.dtype == "int64" and dst_type == "float16":
        x = cast.cast(x, "float32")
+    if x.dtype == "float16" and dst_type == "int32":
+        x = topi.trunc(x)
    return cast.cast(x, dst_type)