[IE][Myriad] fix HW tiling (#894)

This commit is contained in:
Evgeny Latkin
2020-06-11 20:48:56 +03:00
committed by GitHub
parent 30f8af70fc
commit 4fa251483a

View File

@@ -430,6 +430,19 @@ bool checkHWRestrictions(
int kernelSizeX, int kernelSizeY,
int kernelStride,
HwOpMode mode, HwOpType type) {
// Workaround for HW ops failure if too wide input:
// Looks like HW operations (primarily Pooling) can
// use only part of available CMX, up to 1014 * 128
// bits (i.e. 1014 * 16 bytes)
// Provided HwOpMode is 16x16, this means HW needs
// to read up to 16 lines of input tensor, so each
// line mustn't exceed 1014 bytes or 507 pixels if
// precision is FP16
// More details available with the ticket #-33366
if (inTileWidth > 507) {
return false;
}
const int chansPerBlock = 1 << static_cast<int>(mode);
int noOfBlocks = divUp(inTileChannels, chansPerBlock);