| 886 | } |
| 887 | |
| 888 | const std::string versionToStr(int version){ |
| 889 | switch (version) { |
| 890 | case 1: return "f32: No-Op"; |
| 891 | case 2: return "f32: naive matmul"; |
| 892 | case 3: return "f32: tiling"; |
| 893 | case 4: return "f32: 1D blocktiling"; |
| 894 | case 5: return "f32: 2D blocktiling"; |
| 895 | case 6: return "f32: 1D blocktiling with loop unrolling"; |
| 896 | case 7: return "f32: 2D blocktiling with loop unrolling"; |
| 897 | case 8: return "f32: 2D blocktiling with loop unrolling and vectorization"; |
| 898 | case 9: return "f32: 2D blocktiling with loop unrolling, vectorization and transpose"; |
| 899 | case 10: return "f16: 2D blocktiling with loop unrolling and vectorization"; |
| 900 | case 11: return "f16: 2D blocktiling with loop unrolling, vectorization and transpose"; |
| 901 | default: return "Not specified"; |
| 902 | } |
| 903 | } |
| 904 | |
| 905 | int main() { |
| 906 | char* version_str = getenv("MATMUL_VERSION"); |