From 0c64541cbd9174830cbef29d095fdebd51dc29b6 Mon Sep 17 00:00:00 2001 From: Florian Pfaff <6773539+FlorianPfaff@users.noreply.github.com> Date: Wed, 1 Jul 2026 23:21:46 +0200 Subject: [PATCH] Support float16 aliases in PyTorch as_dtype --- src/pyrecest/_backend/pytorch/_dtype.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/pyrecest/_backend/pytorch/_dtype.py b/src/pyrecest/_backend/pytorch/_dtype.py index fdd0fdea1..e70d03efa 100644 --- a/src/pyrecest/_backend/pytorch/_dtype.py +++ b/src/pyrecest/_backend/pytorch/_dtype.py @@ -19,6 +19,7 @@ bool as torch_bool, complex64, complex128, + float16, float32, float64, int8, @@ -37,6 +38,7 @@ "int16": int16, "int32": int32, "int64": int64, + "float16": float16, "float32": float32, "float64": float64, "complex64": complex64,