"examples/controlnet/train_controlnet.py" did not exist on "946d1cb200a875f694818be37c9c9f7547e9db45"
Commit d9716db9 authored by Rosty Geyyer's avatar Rosty Geyyer
Browse files

Add debug info to DeviceGemmXdl

parent d780f778
......@@ -527,6 +527,9 @@ struct DeviceGemmXdl : public DeviceGemm<ALayout,
{
auto str = std::stringstream();
std::map<LoopScheduler, std::string> LoopSchedToString{
{LoopScheduler::Default, "Default"}, {LoopScheduler::Interwave, "Interwave"}};
// clang-format off
str << "DeviceGemmXdl"
<< "<"
......@@ -539,7 +542,13 @@ struct DeviceGemmXdl : public DeviceGemm<ALayout,
<< NPerXDL << ", "
<< MXdlPerWave << ", "
<< NXdlPerWave
<< ">";
<< ">"
<< " NumPrefetch: "
<< NumPrefetch << ", "
<< "LoopScheduler: "
<< LoopSchedToString[LoopSched].c_str() << ", "
<< "PipelineVersion: "
<< PipelineVersion;
// clang-format on
return str.str();
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment