ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-20 04:49:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Patch for bwd data comments (#174)

Browse Source 
* change function name and way to set input zero

* change enable if

[ROCm/composable_kernel commit: 6717168c18]
This commit is contained in:
ltqin
2022-04-05 09:33:53 +08:00
committed by GitHub
parent 8db6e759dc
commit 4012dedcbd
4 changed files with 26 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
profiler/include/profile_convnd_bwd_data_impl.hpp
View File

@@ -336,8 +336,7 @@ bool profile_convnd_bwd_data_impl(int do_verification,
wei_device_buf.ToDevice(weights.mData.data());
// reset input to zero
input_device_result.GenerateTensorValue(GeneratorTensor_1<InDataType>{0});
in_device_buf.ToDevice(input_device_result.mData.data());
in_device_buf.SetZero();
if(do_verification)
{