[FRONTEND] Removed circular import that broke Python 3.6 support (#272)

2021-09-09 13:46:55 -07:00
parent 585e5cd0ec
commit 43723ccb95
4 changed files with 4 additions and 4 deletions
--- a/lib/codegen/transform/prefetch.cc
+++ b/lib/codegen/transform/prefetch.cc
@@ -83,7 +83,7 @@ void prefetch::run(ir::module &mod) {
  }

  // move loads to the beginning of the loop
-  if (tgt_->as_nvidia()->sm() < 80) {
+  if (tgt_->as_nvidia() && tgt_->as_nvidia()->sm() < 80) {
    for (ir::function *fn : mod.get_function_list())
    for (ir::basic_block *bb : fn->blocks()) {
      // only apply to loop body