[IE][Myriad] fix HW tiling (#894)
This commit is contained in:
@@ -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);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user