Fix bugs for different CopyModes
Change-Id: Ie59fb3b36a9ff2de178307a6d5756b5c71e0306b
Этот коммит содержится в:
@@ -189,6 +189,7 @@ void UnpinnedCopyEngine::CopyHostToDevice(UnpinnedCopyEngine::CopyMode copyMode,
|
|||||||
|
|
||||||
if (copyMode == UseMemcpy) {
|
if (copyMode == UseMemcpy) {
|
||||||
|
|
||||||
|
CopyHostToDeviceMemcpy(dst, src, sizeBytes, waitFor);
|
||||||
|
|
||||||
|
|
||||||
} else if (copyMode == UsePinInPlace) {
|
} else if (copyMode == UsePinInPlace) {
|
||||||
@@ -316,7 +317,7 @@ void UnpinnedCopyEngine::CopyDeviceToHost(CopyMode copyMode ,void* dst, const vo
|
|||||||
|
|
||||||
if (copyMode == UsePinInPlace) {
|
if (copyMode == UsePinInPlace) {
|
||||||
CopyDeviceToHostPinInPlace(dst, src, sizeBytes, waitFor);
|
CopyDeviceToHostPinInPlace(dst, src, sizeBytes, waitFor);
|
||||||
} if (copyMode == UseStaging) {
|
} else if (copyMode == UseStaging) {
|
||||||
CopyDeviceToHostStaging(dst, src, sizeBytes, waitFor);
|
CopyDeviceToHostStaging(dst, src, sizeBytes, waitFor);
|
||||||
} else {
|
} else {
|
||||||
// Unknown copy mode.
|
// Unknown copy mode.
|
||||||
|
|||||||
Ссылка в новой задаче
Block a user