Skip to content

Commit

Permalink
Fix format error in CUDA kernel autodiff blog post (#265)
Browse files Browse the repository at this point in the history
  • Loading branch information
kchristin22 authored Nov 5, 2024
1 parent a43e56d commit 8a173e0
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions _posts/2024-11-04-reverse-mode-autodiff-of-cuda-kernels-final.md
Original file line number Diff line number Diff line change
Expand Up @@ -401,15 +401,15 @@ int main() {
const size_type D1 = 2, D2 = 3, D3 = 4, D4 = 3, D5 = 2;

float A[D1][D2][D3] = {
{{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}},
{{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} }
};

float B[D3][D4][D5] = {
{{1, 2}, {3, 4}, {5, 6}},
{{7, 8}, {9, 10}, {11, 12}},
{{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} }
};

float C[D1][D2][D4][D5] = {0}; // Result tensor
Expand Down

0 comments on commit 8a173e0

Please sign in to comment.