[runtime/jit] made auto-tuning silent

2019-07-16 14:41:38 -07:00
parent 7d1797cd32
commit 28959fe165
6 changed files with 105 additions and 172 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});
+  shift.enqueue(&stream, {&a, &b, &c}, true);
  return torchc;
 }