Browse Source

fix(rocm): fix rocm megdnntest sleep and a cut code

GitOrigin-RevId: 26de5ca98b
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
cc952b2b92
2 changed files with 12 additions and 1 deletions
  1. +1
    -1
      dnn/src/rocm/sleep/opr_impl.cpp
  2. +11
    -0
      src/core/impl/utils/persistent_cache.cpp

+ 1
- 1
dnn/src/rocm/sleep/opr_impl.cpp View File

@@ -22,7 +22,7 @@ void SleepForwardImpl::exec() {
double seconds = m_param.time; double seconds = m_param.time;
megdnn_assert(seconds > 0); megdnn_assert(seconds > 0);
auto hdl = static_cast<HandleImpl*>(handle()); auto hdl = static_cast<HandleImpl*>(handle());
sleep(hdl->stream(), hdl->device_prop().clockRate * 18.01 * seconds);
sleep(hdl->stream(), hdl->device_prop().clockRate * 1000 * seconds);
} }


} // namespace rocm } // namespace rocm


+ 11
- 0
src/core/impl/utils/persistent_cache.cpp View File

@@ -122,6 +122,17 @@ std::string PersistentCache::make_category_from_comp_node(CompNode comp_node) {
break; break;
} }
#endif #endif
#if MGB_ROCM
case CompNode::DeviceType::ROCM: {
int drv = -1, hip_rt = -1;
MGB_ROCM_CHECK(hipDriverGetVersion(&drv));
MGB_ROCM_CHECK(hipRuntimeGetVersion(&hip_rt));
auto&& prop = env.rocm_env().device_prop;
return ssprintf("plat=rocm;dev=%s;cap=%d.%d,drv=%d;runtime=%d",
prop.name, prop.major, prop.minor, drv, hip_rt);
break;
}
#endif
case CompNode::DeviceType::CPU: case CompNode::DeviceType::CPU:
return "plat=cpu"; return "plat=cpu";
default: default:


Loading…
Cancel
Save