Floating power!


SUBMITTED BY: Arvamer

DATE: Nov. 14, 2015, 7:35 p.m.

FORMAT: C++

SIZE: 1.5 kB

HITS: 543

  1. #include <iostream>
  2. #include <chrono>
  3. #include <algorithm>
  4. #include <cstdio>
  5. #include <string>
  6. void multiply_tr(FT* m1, FT* m2, size_t size, FT* result);
  7. int main(int argc, char** argv)
  8. {
  9. constexpr size_t dim {4};
  10. constexpr size_t size {dim*dim};
  11. int times = argc > 1 ? std::stoi(argv[1]) : 100;
  12. FT m1[size] = {1.5, 2.3, 1.1, 4,
  13. 1.2, 2.2, 3.3, 4.4,
  14. 3.1, .1, 2.2, .87,
  15. 5.5, 0.8, 1.7, 2.2};
  16. FT res[size];
  17. std::fill(res, res + size, 0);
  18. auto s = std::chrono::high_resolution_clock::now();
  19. for (int i {}; i < times; ++i) multiply_tr(m1, m1, dim, res);
  20. auto e = std::chrono::high_resolution_clock::now();
  21. //std::chrono::duration<std::chrono::nanoseconds> t = e - s;
  22. std::printf("Time to complete: %ldns.\n",
  23. std::chrono::duration_cast<std::chrono::nanoseconds>(e - s).count());
  24. for (size_t i {}; i < dim; ++i)
  25. {
  26. for (size_t j {}; j < dim; ++j)
  27. std::printf("%g, ", res[i*dim + j]);
  28. std::puts("");
  29. }
  30. return 0;
  31. }
  32. void multiply_tr(FT* m1, FT* m2, size_t size, FT* result)
  33. {
  34. for (size_t i {}; i < size; ++i)
  35. for (size_t j {}; j < size; ++j)
  36. {
  37. auto pos_row1 = i * size;
  38. auto pos_row2 = j * size;
  39. auto pos_result = i * size + j;
  40. for (size_t k {}; k < size; ++k)
  41. result[pos_result + k] += m1[pos_row1] * m2[pos_row2];
  42. }
  43. }

comments powered by Disqus