[Bugfix] Use reverse_cache_read for spmm example (#92)

* use reverse_cache_read for spmm example * remove redundancy * docker
uwsampl · Mar 6, 2023 · f0dac6d · f0dac6d
1 parent b23f4cd
commit f0dac6d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docker/Dockerfile.ci_sparsetir_gpu b/docker/Dockerfile.ci_sparsetir_gpu
@@ -78,4 +78,4 @@ RUN bash docker/install/install_sparsetir_gpu.sh
 
 # Install dependencies required by lint
 RUN pip3 install flake8==5.0.4 pylint==2.15.6 cpplint==1.6.1 black==22.8.0
-RUN apt install -y default-jre
+RUN apt-get update --fix-missing && apt-get install -y default-jre
diff --git a/examples/spmm/bench_spmm.py b/examples/spmm/bench_spmm.py
@@ -173,7 +173,7 @@ def bench_hyb(
             sch.reorder(foo, fi, j, foi)
             if is_atomic:
                 sch.annotate(blk, "atomic", True)
-                write_blk = sch.cache_write(blk, 0, "local")
+                write_blk = sch.reverse_cache_write(blk, 0, "local")
                 sch.reverse_compute_at(write_blk, fi, True)
                 # sch.unroll(sch.get_loops(write_blk)[-2])
             sch.bind(fi, "threadIdx.x")