We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a43e56d commit 2164634Copy full SHA for 2164634
_posts/2024-11-04-reverse-mode-autodiff-of-cuda-kernels-final.md
@@ -401,15 +401,15 @@ int main() {
401
const size_type D1 = 2, D2 = 3, D3 = 4, D4 = 3, D5 = 2;
402
403
float A[D1][D2][D3] = {
404
- {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}},
405
- {{13, 14, 15, 16}, {17, 18, 19, 20}, {21, 22, 23, 24}}
+ { {1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12} },
+ { {13, 14, 15, 16}, {17, 18, 19, 20}, {21, 22, 23, 24} }
406
};
407
408
float B[D3][D4][D5] = {
409
- {{1, 2}, {3, 4}, {5, 6}},
410
- {{7, 8}, {9, 10}, {11, 12}},
411
- {{13, 14}, {15, 16}, {17, 18}},
412
- {{19, 20}, {21, 22}, {23, 24}}
+ { {1, 2}, {3, 4}, {5, 6} },
+ { {7, 8}, {9, 10}, {11, 12} },
+ { {13, 14}, {15, 16}, {17, 18} },
+ { {19, 20}, {21, 22}, {23, 24} }
413
414
415
float C[D1][D2][D4][D5] = {0}; // Result tensor
0 commit comments