[python] upgraded pybind11 ; forcing torch tensors to be contiguous()

2019-09-05 12:30:51 -04:00
parent 58544d0523
commit 2d6c8311e8
23 changed files with 960 additions and 531 deletions
--- a/python/triton/kernel.py
+++ b/python/triton/kernel.py
@@ -217,6 +217,7 @@ class kernel:
    if fw.has_tensorflow():
      return self.fw_op(*op_args, id=op_id)
    elif fw.has_torch():
-      return self.fw_op(op_id, *op_args)
+      args = [x.contiguous() if isinstance(x, fw.torch.Tensor) else x for x in op_args]
+      return self.fw_op(op_id, *args)
    else:
      assert False