gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[dnn/shift] now strictly only shifting the interior

Browse Source
This commit is contained in:
Philippe Tillet
2019-07-16 20:18:48 -07:00
parent ec24e1e7df
commit 07c964919c
6 changed files with 34 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/python/pytorch/shift.cpp
View File

@@ -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;
}

2
examples/python/tensorflow/shift.cpp
View File

@@ -122,7 +122,7 @@ public:
triton::driver::cu_buffer da(ctx, (CUdeviceptr)tf_a.flat<Eigen::half>().data(), false);
triton::driver::cu_buffer db(ctx, (CUdeviceptr)tf_b.flat<Eigen::half>().data(), false);
triton::driver::cu_buffer dc(ctx, (CUdeviceptr)tf_c->flat<Eigen::half>().data(), false);
shift.enqueue(stream, {&da, &db, &dc});
shift.enqueue(stream, {&da, &db, &dc}, false);
}
private: