diff --git a/lib/codegen/analysis/swizzle.cc b/lib/codegen/analysis/swizzle.cc index 304aac95a..fcde938d9 100644 --- a/lib/codegen/analysis/swizzle.cc +++ b/lib/codegen/analysis/swizzle.cc @@ -30,7 +30,7 @@ void swizzle::run(ir::module &) { if(!in_layout) continue; int dtsize = layout->get_type()->get_scalar_ty()->get_primitive_size_in_bits() / 8; - if(tgt_->as_nvidia()->sm() < 80){ + if(tgt_->as_nvidia() && tgt_->as_nvidia()->sm() < 80){ int inner = mma_dot_a ? 0 : 1; per_phase_[layout] = std::max(128 / (in_layout->mts(ord[0])*in_layout->nts(ord[0])*dtsize), 1); max_phase_[layout] = (ord[inner] == 1 ? 8 : 4) / per_phase_[layout]; diff --git a/lib/codegen/transform/peephole.cc b/lib/codegen/transform/peephole.cc index 043e64c05..1c8920230 100644 --- a/lib/codegen/transform/peephole.cc +++ b/lib/codegen/transform/peephole.cc @@ -285,7 +285,7 @@ void peephole::run(ir::module &mod) { was_modified = was_modified || rewrite_gep_ptr_min_off_plus_off(i, builder); was_modified = was_modified || rewrite_select_masked_load(i, builder); was_modified = was_modified || rewrite_cvt_layout(i, builder); - if(tgt_->as_nvidia()->sm() >= 80) + if(tgt_->as_nvidia() && tgt_->as_nvidia()->sm() >= 80) was_modified = was_modified || rewrite_load_to_shared(i, builder); if(was_modified) seen.insert(i); diff --git a/lib/codegen/transform/prefetch.cc b/lib/codegen/transform/prefetch.cc index e4e32b2ec..c54a92005 100644 --- 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 diff --git a/python/triton/language/random.py b/python/triton/language/random.py index 913073679..de9c1b3b2 100644 --- a/python/triton/language/random.py +++ b/python/triton/language/random.py @@ -1,5 +1,5 @@ import triton -import triton.language as tl +from . import core as tl # Notes