"git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "4c8fd7608e564ae5c4d5c72432262a1bcf8e54f3"
Commit a0049dde authored by Bartlomiej Kocot's avatar Bartlomiej Kocot
Browse files

Log CDEBlockTransferScalarPerVector_NPerBlock in conv_fwd_multi_d_xdl implementation

parent 68f2b5e7
...@@ -912,6 +912,7 @@ struct DeviceGroupedConvFwdMultipleD_Xdl_CShuffle ...@@ -912,6 +912,7 @@ struct DeviceGroupedConvFwdMultipleD_Xdl_CShuffle
<< NXdlPerWave << ", " << NXdlPerWave << ", "
<< ABlockTransferSrcScalarPerVector << ", " << ABlockTransferSrcScalarPerVector << ", "
<< BBlockTransferSrcScalarPerVector << ", " << BBlockTransferSrcScalarPerVector << ", "
<< CDEBlockTransferScalarPerVector_NPerBlock << ", "
<< CShuffleMXdlPerWavePerShuffle << ", " << CShuffleMXdlPerWavePerShuffle << ", "
<< CShuffleNXdlPerWavePerShuffle << CShuffleNXdlPerWavePerShuffle
<< ">"; << ">";
......
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