[BACKEND] Added Int8 mma (#440)

2022-01-28 01:12:44 +08:00
parent 3a23c1dd33
commit 59d371c6eb
11 changed files with 232 additions and 115 deletions
--- a/lib/codegen/analysis/swizzle.cc
+++ b/lib/codegen/analysis/swizzle.cc
@@ -41,9 +41,15 @@ void swizzle::run(ir::module &) {
          vec_[layout] = 2*layouts_->get(mma_dot_b)->to_mma()->rep(1);
      }
      else {
+        if (!layout->allow_swizzle()) {
+          per_phase_[layout] = 1;
+          max_phase_[layout] = 1;
+          vec_[layout] = 1;
+        } else {
          per_phase_[layout] = std::max<int>(128 / (in_layout->mts(ord[0])*in_layout->nts(ord[0])*dtsize), 1);
          max_phase_[layout] = layout->get_mma_strided() / per_phase_[layout];
          vec_[layout]       = layout->get_mma_vec();
+        }
      }
    }
 }