From 277b712284300088f32fc6d1b3ab1b8fae14a91a Mon Sep 17 00:00:00 2001 From: Michael Melesse Date: Mon, 31 Oct 2022 19:11:58 +0000 Subject: [PATCH] save changes --- lib/codegen/selection/generator.cc | 12 ++++++------ scripts/amd/backtrace.sh | 2 +- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/lib/codegen/selection/generator.cc b/lib/codegen/selection/generator.cc index d35245942..3335156ab 100644 --- a/lib/codegen/selection/generator.cc +++ b/lib/codegen/selection/generator.cc @@ -2974,18 +2974,18 @@ void generator::visit_dot_inst(ir::dot_inst* dot) { unsigned NK = A_shapes[red_axis]; bool is_outer = NK == 1; #ifdef USE_ROCM - bool is_mma = layouts_->get(dot)->to_mma(); -#else bool is_mma = false; +#else + bool is_mma = layouts_->get(dot)->to_mma(); #endif if(!is_outer && is_mma && tgt_->as_nvidia() && tgt_->as_nvidia()->sm() < 80) return visit_mma884(dot, A, B, D, NK); if(!is_outer && is_mma && tgt_->as_nvidia() && tgt_->as_nvidia()->sm() >= 80) return visit_mma16816(dot, A, B, D, NK); // rename it as visit_mma_v2()? - if (dot->get_type()->get_scalar_ty()->is_fp32_ty() && - A->get_type()->get_scalar_ty()->is_fp32_ty()) - return visit_fmadot(dot, A, B, D, NK, c_ty, f_mul_add); - throw std::runtime_error("dot has invalid operand type"); + // if (dot->get_type()->get_scalar_ty()->is_fp32_ty() && + // A->get_type()->get_scalar_ty()->is_fp32_ty()) + return visit_fmadot(dot, A, B, D, NK, c_ty, f_mul_add); + // throw std::runtime_error("dot has invalid operand type"); } void generator::visit_trans_inst(ir::trans_inst* trans) { diff --git a/scripts/amd/backtrace.sh b/scripts/amd/backtrace.sh index 41d66f985..1616f481e 100644 --- a/scripts/amd/backtrace.sh +++ b/scripts/amd/backtrace.sh @@ -7,7 +7,7 @@ sudo apt install gdb -y gdb -ex "set pagination off" \ -ex "file python" \ - -ex 'run -m pytest --capture=tee-sys --verbose "python/test/unit/language/test_core.py::test_dot"' \ + -ex 'run -m pytest --capture=tee-sys --verbose "python/test/unit/language/test_core.py::test_dot[none-False-float16]"' \ -ex "backtrace" \ -ex "set confirm off" \ -ex "q" \