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

[dnn/conv] fixed formatting of generated Triton-C code

Browse Source
This commit is contained in:
Philippe Tillet
2019-05-16 15:48:02 -04:00
parent ece7beea3c
commit 34f8617709
2 changed files with 115 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/cpp/conv.cpp
View File

@@ -12,9 +12,9 @@ int main() {
triton::jit jit(context);
triton::dnn::conv::type ty = triton::dnn::conv::WGRAD;
// initialization
int32_t B = 4, NF = 32;
int32_t D = 1, H = 24, W = 240;
int32_t NC = 32, T = 1, R = 3, S = 3;
int32_t B = 32, NF = 128;
int32_t D = 1, H = 56, W = 56;
int32_t NC = 128, T = 1, R = 3, S = 3;
int32_t pad_d = 0, pad_h = 1, pad_w = 1;
triton::dnn::conv configuration(B, NC, D, H, W, T, R, S, NF, 1, 1, 1, pad_d, pad_h, pad_w, ty);
// convolution configuration