fix: extend f32 matmul precision to ROCm for Qwen-Image, Krea2 and Boogu (#1772)

This commit is contained in:
Piotr Wilkin (ilintar) authored and GitHub committed 2026-07-10 23:23:01 +08:00
1 parent 9beb6aca69
commit ead6bf521b
4 files changed
+6 -6

No files matched your search

+1 -1
View File
@@ -267,7 +267,7 @@ namespace Krea2 {
auto knorm = std::dynamic_pointer_cast<KreaRMSNorm>(blocks["qknorm.knorm"]);
auto wo = std::dynamic_pointer_cast<Linear>(blocks["wo"]);
if (sd_backend_is(ctx->backend, "Vulkan")) {
if (sd_backend_is(ctx->backend, "Vulkan") || sd_backend_is(ctx->backend, "ROCm")) {
wo->set_force_prec_f32(true);
}