diff options
Diffstat (limited to 'src/backends/reference/RefWorkloadFactory.cpp')
| -rw-r--r-- | src/backends/reference/RefWorkloadFactory.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp index c9fc626bc1..5d3775a59d 100644 --- a/src/backends/reference/RefWorkloadFactory.cpp +++ b/src/backends/reference/RefWorkloadFactory.cpp @@ -194,6 +194,13 @@ std::unique_ptr<IWorkload> RefWorkloadFactory::CreateConvertFp16ToFp32( return std::make_unique<RefConvertFp16ToFp32Workload>(descriptor, info); } +std::unique_ptr<IWorkload> RefWorkloadFactory::CreateConvertFp32ToBf16( + const ConvertFp32ToBf16QueueDescriptor& descriptor, + const WorkloadInfo& info) const +{ + return std::make_unique<RefConvertFp32ToBf16Workload>(descriptor, info); +} + std::unique_ptr<IWorkload> RefWorkloadFactory::CreateConvertFp32ToFp16( const ConvertFp32ToFp16QueueDescriptor& descriptor, const WorkloadInfo& info) const |