[dnn/shift] now strictly only shifting the interior

2019-07-16 20:18:48 -07:00
parent ec24e1e7df
commit 07c964919c
6 changed files with 34 additions and 35 deletions
--- a/examples/python/pytorch/shift.cpp
+++ b/examples/python/pytorch/shift.cpp
@@ -75,7 +75,7 @@ torch::Tensor shift_common(

  triton::driver::cu_buffer c(ctx, (CUdeviceptr)torchc.storage().data(), false);
  // Enqueue
-  shift.enqueue(&stream, {&a, &b, &c}, true);
+  shift.enqueue(&stream, {&a, &b, &c}, false);
  return torchc;
 }