experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix a bug in device tracer.

Browse Source 
PiperOrigin-RevId: 243477024
This commit is contained in:
Xiao Yu 2019-04-14 00:33:00 -07:00 committed by TensorFlower Gardener
parent 256add6a5b
commit 0219034f33
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/platform/default/device_tracer.cc
View File

@ -231,7 +231,7 @@ class CuptiCallbackHook {
src_type, dst_type, params->ByteCount, cbdata.context, nullptr); src_type, dst_type, params->ByteCount, cbdata.context, nullptr);
} }
template <typename T> template <typename T>
static void StartMemcpyAsync(CUmemorytype dst_type, CUmemorytype src_type, static void StartMemcpyAsync(CUmemorytype src_type, CUmemorytype dst_type,
const CUpti_CallbackData& cbdata, const CUpti_CallbackData& cbdata,
CudaEventRecorder* recorder) { CudaEventRecorder* recorder) {
auto params = static_cast<const T*>(cbdata.functionParams); auto params = static_cast<const T*>(cbdata.functionParams);