mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-21 01:04:57 +08:00
25a5ed0dca
PVC platform with no support for atomic operations on system memory must always allocate buffers in local memory to avoid atomic access violation. Note: the feature is being implemented under the new registry key AllocateBuffersInLocalMemoryForMultiRootDeviceContexts (disabled by default) Related-To: NEO-7092 Signed-off-by: Milczarek, Slawomir <slawomir.milczarek@intel.com>
38 lines
1.2 KiB
C++
38 lines
1.2 KiB
C++
/*
|
|
* Copyright (C) 2021-2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/utilities/reference_tracked_object.h"
|
|
|
|
#include <cstdint>
|
|
#include <limits>
|
|
|
|
namespace NEO {
|
|
class MigrationSyncData : public ReferenceTrackedObject<MigrationSyncData> {
|
|
public:
|
|
static constexpr uint32_t locationUndefined = std::numeric_limits<uint32_t>::max();
|
|
|
|
MigrationSyncData(size_t size);
|
|
~MigrationSyncData() override;
|
|
uint32_t getCurrentLocation() const;
|
|
void startMigration();
|
|
void setCurrentLocation(uint32_t rootDeviceIndex);
|
|
MOCKABLE_VIRTUAL void signalUsage(volatile uint32_t *tagAddress, uint32_t taskCount);
|
|
bool isUsedByTheSameContext(volatile uint32_t *tagAddress) const;
|
|
MOCKABLE_VIRTUAL void waitOnCpu();
|
|
bool isMigrationInProgress() const { return migrationInProgress; }
|
|
void *getHostPtr() const { return hostPtr; }
|
|
|
|
protected:
|
|
MOCKABLE_VIRTUAL void yield() const;
|
|
volatile uint32_t *tagAddress = nullptr;
|
|
void *hostPtr = nullptr;
|
|
uint32_t latestTaskCountUsed = 0u;
|
|
uint32_t currentLocation = locationUndefined;
|
|
bool migrationInProgress = false;
|
|
};
|
|
} // namespace NEO
|