Fix packet count when programming L3 flush

Related-To: NEO-5968

Signed-off-by: Aravind Gopalakrishnan <aravind.gopalakrishnan@intel.com>
This commit is contained in:
Aravind Gopalakrishnan
2022-04-17 01:13:30 +00:00
committed by Compute-Runtime-Automation
parent e47385dc11
commit 79b8051eab
4 changed files with 10 additions and 2 deletions

View File

@@ -58,6 +58,7 @@ struct Event : _ze_event_handle_t {
virtual uint64_t getGpuAddress(Device *device) = 0;
virtual uint32_t getPacketsInUse() = 0;
virtual uint32_t getPacketsUsedInLastKernel() = 0;
virtual uint64_t getPacketAddress(Device *device) = 0;
virtual void resetPackets() = 0;
void *getHostAddress() { return hostAddress; }
@@ -168,6 +169,7 @@ struct EventImp : public Event {
void resetPackets() override;
uint64_t getPacketAddress(Device *device) override;
uint32_t getPacketsInUse() override;
uint32_t getPacketsUsedInLastKernel() override;
void setPacketsInUse(uint32_t value) override;
std::unique_ptr<KernelEventCompletionData<TagSizeT>[]> kernelEventCompletionData;

View File

@@ -380,6 +380,11 @@ uint32_t EventImp<TagSizeT>::getPacketsInUse() {
return packetsInUse;
}
template <typename TagSizeT>
uint32_t EventImp<TagSizeT>::getPacketsUsedInLastKernel() {
return kernelEventCompletionData[getCurrKernelDataIndex()].getPacketsUsed();
}
template <typename TagSizeT>
void EventImp<TagSizeT>::setPacketsInUse(uint32_t value) {
kernelEventCompletionData[getCurrKernelDataIndex()].setPacketsUsed(value);