Skip to content

GitLab

Commit c6e888bd authored by qisan's avatar qisan
Browse files

Merge: async_copy & lds_copy

parent 32d0b3cb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/target/codegen_hip.cc
View file @ c6e888bd
...@@ -1099,7 +1099,7 @@ void CodeGenTileLangHIP::VisitExpr_(const CallNode *op, std::ostream &os) { ...@@ -1099,7 +1099,7 @@ void CodeGenTileLangHIP::VisitExpr_(const CallNode *op, std::ostream &os) {
} else { } else {
offset = this->PrintExpr(op->args[1]); offset = this->PrintExpr(op->args[1]);
} }
os << "make_wave_buffer_resource(" << base_ptr << " + (" << offset << "))"; os << "tl::make_wave_buffer_resource(" << base_ptr << " + (" << offset << "))";
} }
else if (op->op.same_as(Op::Get("tl.dcu_async_copy"))) { else if (op->op.same_as(Op::Get("tl.dcu_async_copy"))) {
...@@ -1124,7 +1124,7 @@ void CodeGenTileLangHIP::VisitExpr_(const CallNode *op, std::ostream &os) { ...@@ -1124,7 +1124,7 @@ void CodeGenTileLangHIP::VisitExpr_(const CallNode *op, std::ostream &os) {
// 3. 仿照范例进行流输出 // 3. 仿照范例进行流输出
this->PrintIndent(); this->PrintIndent();
this->stream << "cp_async_gs<" this->stream << "tl::cp_async_gs<"
<< N << ", " << N << ", "
<< smem_offset << ", " << smem_offset << ", "
<< load_count << ", " << load_count << ", "
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment