sgemm_wg: Output CPU data to binary

2024-04-14 21:25:14 -07:00
parent 4e9855dc33
commit 6cbfbfb856
2 changed files with 8 additions and 1 deletions
--- a/tests/regression/sgemm_wg/main.cpp
+++ b/tests/regression/sgemm_wg/main.cpp
@@ -119,6 +119,14 @@ int run_test(const kernel_arg_t& kernel_arg,
  file.write(reinterpret_cast<char *>(staging_buf.data()), buf_size);
  file.close();

+  std::ofstream ref_file("reference.c.bin", std::ios::binary | std::ios::out);
+  if (!ref_file) {
+    std::cerr << "error: failed to open reference.c.bin for writing\n";
+    exit(EXIT_FAILURE);
+  }
+  ref_file.write(reinterpret_cast<char *>(ref_data.data()), buf_size);
+  ref_file.close();
+
  // verify result
  std::cout << "verify result" << std::endl;
  {