* Do not remove convert after the topK * Added debug message * Removed xFail * Revert "Added debug message" This reverts commit a01ace4ade88d73e2797b47c58db33943b0f508d. * Added test
* Do not remove convert after the topK * Added debug message * Removed xFail * Revert "Added debug message" This reverts commit a01ace4ade88d73e2797b47c58db33943b0f508d. * Added test