feat(hip): use hostToDevice/deviceToHost timers

src/hip/fasten.hpp

@@ -215,7 +215,7 @@ template <size_t PPWI> class IMPL_CLS final : public Bude<PPWI> {
 
     Sample sample(PPWI, wgsize, p.nposes());
 
-    auto contextStart = now();
+    auto hostToDeviceStart = now();
     auto protein = allocate(p.protein);
     auto ligand = allocate(p.ligand);
     auto transforms_0 = allocate(p.poses[0]);
@@ -227,9 +227,9 @@ template <size_t PPWI> class IMPL_CLS final : public Bude<PPWI> {
     auto forcefield = allocate(p.forcefield);
     auto results = allocate<float>(sample.energies.size());
     checkError(hipDeviceSynchronize());
-    auto contextEnd = now();
+    auto hostToDeviceEnd = now();
 
-    sample.contextTime = {contextStart, contextEnd};
+    sample.hostToDevice = {hostToDeviceStart, hostToDeviceEnd};
 
     size_t global = std::ceil(double(p.nposes()) / PPWI);
     global = std::ceil(double(global) / double(wgsize));
@@ -247,8 +247,12 @@ template <size_t PPWI> class IMPL_CLS final : public Bude<PPWI> {
       sample.kernelTimes.emplace_back(kernelStart, kernelEnd);
     }
 
+    auto deviceToHostStart = now();
     checkError(
         hipMemcpy(sample.energies.data(), results, sample.energies.size() * sizeof(float), hipMemcpyDeviceToHost));
+    auto deviceToHostEnd = now();
+
+    sample.deviceToHost = {deviceToHostStart, deviceToHostEnd};
 
     free(protein);
     free(ligand);