Skip to content

Commit

Permalink
[Driver] Fix LogCmdRedux() for double reduction (#1072)
Browse files Browse the repository at this point in the history
* Update LogCmdRedux() for double

* Fix the print string in reduce_driver.cpp
qianfengz authored Aug 2, 2021

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent c701af3 commit 74193e9
Showing 2 changed files with 4 additions and 2 deletions.
4 changes: 2 additions & 2 deletions driver/reduce_driver.hpp
Original file line number Diff line number Diff line change
@@ -442,9 +442,9 @@ int ReduceDriver<Tgpu, Tref>::RunForwardGPU()

STOP_TIME
if(WALL_CLOCK)
printf("Wall-clock Time Forward LRN Elapsed: %f ms\n",
printf("Wall-clock Time Reduction Elapsed: %f ms\n",
t.gettime_ms() / inflags.GetValueInt("iter"));
printf("GPU Kernel Time Forward LRN Elapsed: %f ms\n", time);
printf("GPU Kernel Time Reduction Elapsed: %f ms\n", time);
}

return miopenStatusSuccess;
2 changes: 2 additions & 0 deletions src/reducetensor_api.cpp
Original file line number Diff line number Diff line change
@@ -45,6 +45,8 @@ static void LogCmdRedux(const miopen::ReduceTensorDescriptor reduceTensorDesc,
ss << "reducebfp16";
else if(aDesc.GetType() == miopenInt8 || aDesc.GetType() == miopenInt8x4)
ss << "reduceint8";
else if(aDesc.GetType() == miopenDouble)
ss << "reducefp64";
else
ss << "reduce";

0 comments on commit 74193e9

Please sign in to comment.