Add sub-device test for PARTITION_BY_COUNTS partition type

Fix bugs in OCL stub related with this type of partitioning.