[codegen/alignment_info] better handling of constants

2019-07-18 16:12:06 -07:00
parent 86f70f8224
commit f0d8306437
11 changed files with 57 additions and 33 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}, false);
+  shift.enqueue(&stream, {&a, &b, &c}, true);
  return torchc;
 }