refactor: reduce TBX download timeout for unit tests

Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:
Bartosz Dunajski
2024-09-09 15:31:32 +00:00
committed by Compute-Runtime-Automation
parent eec841b662
commit dd8460beba
3 changed files with 9 additions and 2 deletions

View File

@@ -31,6 +31,9 @@ class TbxCommandStreamReceiverHw : public CommandStreamReceiverSimulatedHw<GfxFa
uint32_t getMaskAndValueForPollForCompletion() const;
bool getpollNotEqualValueForPollForCompletion() const;
void flushSubmissionsAndDownloadAllocations(TaskCountType taskCount, bool skipAllocationsDownload);
MOCKABLE_VIRTUAL uint64_t getNonBlockingDownloadTimeoutMs() const {
return 2000; // 2s
}
public:
using CommandStreamReceiverSimulatedCommonHw<GfxFamily>::initAdditionalMMIO;