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

[codegen/alignment_info] better handling of constants

Browse Source
This commit is contained in:
Philippe Tillet
2019-07-18 16:12:06 -07:00
parent 86f70f8224
commit f0d8306437
11 changed files with 57 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/cpp/shift.cpp
View File

@@ -14,13 +14,13 @@ int main() {
// initialize default compute device
auto context = triton::driver::backend::contexts::get_default();
auto op = triton::dnn::shift::FPROP;
auto op = triton::dnn::shift::WGRAD;
// initialization
int32_t R = 3, S = 3;
int32_t B = 16, F = 4096;
int32_t B = 128, F = 128;
int32_t H = 16, W = 16;
int32_t C = 4096;
int32_t C = 128;
// random shifts
std::vector<int32_t> shift_h(C);