summaryrefslogtreecommitdiff
path: root/test-chill/unit-tests/cpp_validate_prog/mm_three_basic.cc
diff options
context:
space:
mode:
Diffstat (limited to 'test-chill/unit-tests/cpp_validate_prog/mm_three_basic.cc')
-rw-r--r--test-chill/unit-tests/cpp_validate_prog/mm_three_basic.cc33
1 files changed, 33 insertions, 0 deletions
diff --git a/test-chill/unit-tests/cpp_validate_prog/mm_three_basic.cc b/test-chill/unit-tests/cpp_validate_prog/mm_three_basic.cc
new file mode 100644
index 0000000..49df049
--- /dev/null
+++ b/test-chill/unit-tests/cpp_validate_prog/mm_three_basic.cc
@@ -0,0 +1,33 @@
+/*
+<test name=small define="{'AN':2, 'AMBN':5, 'BM':3}">
+ procedure void mm(
+ in float[AN][AMBN] A = matrix([,],lambda i,j: i*AMBN + j),
+ in float[AMBN][BM] B = matrix([,],lambda i,j: i*BM + j),
+ out float[AN][BM] C = matrix([,],lambda i,j: 0))
+</test>
+
+<test name=medium define="{'AN':20, 'AMBN':50, 'BM':30}">
+ procedure void mm(
+ in float[AN][AMBN] A = matrix([,],lambda i,j: i*AMBN + j),
+ in float[AMBN][BM] B = matrix([,],lambda i,j: i*BM + j),
+ out float[AN][BM] C = matrix([,],lambda i,j: 0))
+</test>
+
+<test name=big define="{'AN':200, 'AMBN':500, 'BM':300}">
+ procedure void mm(
+ in float[AN][AMBN] A = matrix([,],lambda i,j: i*AMBN + j),
+ in float[AMBN][BM] B = matrix([,],lambda i,j: i*BM + j),
+ out float[AN][BM] C = matrix([,],lambda i,j: 0))
+</test>
+*/
+
+void mm(float A[AN][AMBN], float B[AMBN][BM], float C[AN][BM]) {
+ for(int i = 0; i < AN; i++) {
+ for(int j = 0; j < BM; j++) {
+ C[i][j] = 0;
+ for(int k = 0; k < AMBN; k++) {
+ C[i][j] += A[i][k] * B[k][j];
+ }
+ }
+ }
+}