gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CORE][TOOLS][BENCH] Disabled normalization for auto-tuning

Browse Source
This commit is contained in:
Philippe Tillet
2020-03-06 18:41:06 -05:00
committed by Philippe Tillet
parent 9cb3fd899a
commit eadaeab299
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/triton/tools/bench.hpp
View File

@@ -41,9 +41,9 @@ inline double bench(std::function<void()> const & op, driver::stream * stream, b
while(total_time*1e-9 < 1e-2){
float norm = 1;
// normalize clock if possible to reduce noise in auto-tuning
if(normalize)
if(auto cu_device = dynamic_cast<const triton::driver::cu_device*>(stream->context()->device()))
norm = (float)cu_device->current_sm_clock()/cu_device->max_sm_clock();
// if(normalize)
// if(auto cu_device = dynamic_cast<const triton::driver::cu_device*>(stream->context()->device()))
// norm = (float)cu_device->current_sm_clock()/cu_device->max_sm_clock();
tmr.start();
op();
stream->synchronize();