scripts/vsmlrt.py: add missing fp16_blacklist_ops to Backend.OV_GPU

Signed-off-by: akarin <[email protected]>
AmusementClub · Sep 23, 2022 · be5fc73 · be5fc73
1 parent c3350f1
commit be5fc73
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/vsmlrt.py b/scripts/vsmlrt.py
@@ -114,7 +114,9 @@ class OV_GPU:
         fp16: bool = False
         num_streams: typing.Union[int, str] = 1
         device_id: int = 0
+        fp16_blacklist_ops: typing.Optional[typing.Sequence[str]] = None
 
+        # internal backend attributes
         supports_onnx_serialization: bool = True
 
     @dataclass(frozen=False)