diff --git a/juno_samples/matmul/src/matmul.sch b/juno_samples/matmul/src/matmul.sch
index b6667596bf6161a7d2d7b9be361155d19087e9bc..6867576e4a3a81f327d23eca6dccbcd6a6183eb4 100644
--- a/juno_samples/matmul/src/matmul.sch
+++ b/juno_samples/matmul/src/matmul.sch
@@ -52,6 +52,8 @@ if feature("cuda") {
   infer-schedules(*);
   dce(*);
   //rewrite(*);
+  let out = outline(matmul@outer);
+  gpu(out);
   fixpoint {
     simplify-cfg(*);
     dce(*);