Allow fill global buffer (#774)

* Allow fill global buffer * fix lint error

Allow fill global buffer (#774)
* Allow fill global buffer * fix lint error
03f21987 · Kurisu · GitHub · 9a869396 · 03f21987
Unverified Commit 03f21987 authored Sep 01, 2025 by Kurisu Committed by GitHub Sep 01, 2025
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/op/elem.cc src/op/elem.cc +2 -1

No files found.
--- a/src/op/elem.cc
+++ b/src/op/elem.cc
@@ -186,7 +186,8 @@ Stmt FillNode::Lower(const LowerArgs &T, arith::Analyzer *analyzer) const {
    auto init_loop = MakeSIMTLoop(analyzer);
    auto vectorized_thread_loop = VectorizeLoop(init_loop);
    return vectorized_thread_loop;
-  } else if (dst.scope() == "shared.dyn" || dst.scope() == "shared") {
+  } else if (dst.scope() == "shared.dyn" || dst.scope() == "shared" ||
+             dst.scope() == "global") {
    auto par_op = ParallelOp(MakeSIMTLoop(analyzer));
    par_op->InferLayout({T.target, T.thread_bounds, T.layout_map},
                        InferLevel::kFree);