Add int32 to fused types with lowered scalarspervector

707d6261 · Alan Turner · d7173bc6 · 707d6261
Commit 707d6261 authored Jun 16, 2023 by Alan Turner
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

library/src/jit_library/src/device_gemm_multiple_d.cpp library/src/jit_library/src/device_gemm_multiple_d.cpp +5 -0

No files found.
--- a/library/src/jit_library/src/device_gemm_multiple_d.cpp
+++ b/library/src/jit_library/src/device_gemm_multiple_d.cpp
@@ -111,6 +111,11 @@ Solution Problem::MakeSolution(std::size_t idx, const std::string& arch) const
        {
            params[params.size() - 3] = "4";
        }
+        if(EDataType == DataType::Int32 or std::any_of(
+               DsDataType.begin(), DsDataType.end(), [](auto t) { return t == DataType::Int32; }))
+        {
+            params[params.size() - 3] = "4";
+        }
    }

    params[a_elementwise_op_idx]  = AElementOp;