ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-08-03 14:18:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix auto_cast matrix bug

Browse Source
This commit is contained in:
gingerBill
2024-12-01 11:54:52 +11:00
committed by flysand7
parent d5b6c25686
commit fafac1bc77
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/llvm_backend_expr.cpp
+12 -1
View File
@@ -2363,12 +2363,23 @@ gb_internal lbValue lb_emit_conv(lbProcedure *p, lbValue value, Type *t) {
GB_ASSERT(src->kind == Type_Matrix); GB_ASSERT(src->kind == Type_Matrix);
lbAddr v = lb_add_local_generated(p, t, true); lbAddr v = lb_add_local_generated(p, t, true);
if (is_matrix_square(dst) && is_matrix_square(dst)) { if (dst->Matrix.row_count == src->Matrix.row_count &&
dst->Matrix.column_count == src->Matrix.column_count) {
for (i64 j = 0; j < dst->Matrix.column_count; j++) {
for (i64 i = 0; i < dst->Matrix.row_count; i++) {
lbValue d = lb_emit_matrix_epi(p, v.addr, i, j);
lbValue s = lb_emit_matrix_ev(p, value, i, j);
s = lb_emit_conv(p, s, dst->Matrix.elem);
lb_emit_store(p, d, s);
}
}
} else if (is_matrix_square(dst) && is_matrix_square(dst)) {
for (i64 j = 0; j < dst->Matrix.column_count; j++) { for (i64 j = 0; j < dst->Matrix.column_count; j++) {
for (i64 i = 0; i < dst->Matrix.row_count; i++) { for (i64 i = 0; i < dst->Matrix.row_count; i++) {
if (i < src->Matrix.row_count && j < src->Matrix.column_count) { if (i < src->Matrix.row_count && j < src->Matrix.column_count) {
lbValue d = lb_emit_matrix_epi(p, v.addr, i, j); lbValue d = lb_emit_matrix_epi(p, v.addr, i, j);
lbValue s = lb_emit_matrix_ev(p, value, i, j); lbValue s = lb_emit_matrix_ev(p, value, i, j);
s = lb_emit_conv(p, s, dst->Matrix.elem);
lb_emit_store(p, d, s); lb_emit_store(p, d, s);
} else if (i == j) { } else if (i == j) {
lbValue d = lb_emit_matrix_epi(p, v.addr, i, j); lbValue d = lb_emit_matrix_epi(p, v.addr, i, j);