NNERuntimeNPU.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "NNERuntimeRunSync.h"
#include "NNEStatus.h"
#include "UObject/Interface.h"
 
#include "NNERuntimeNPU.generated.h"
 
class UNNEModelData;
 
namespace UE::NNE
{
 
class IModelInstanceNPU : public IModelInstanceRunSync
{
};
 
class IModelNPU
{
public:
 
    virtual ~IModelNPU() = default;
 
    virtual TSharedPtr<UE::NNE::IModelInstanceNPU> CreateModelInstanceNPU() = 0;
};
 
} // UE::NNE
 
UINTERFACE(MinimalAPI)
class UNNERuntimeNPU : public UInterface
{
    GENERATED_BODY()
};
 
class INNERuntimeNPU
{
    GENERATED_BODY()
    
public:
 
    using ECanCreateModelNPUStatus = UE::NNE::EResultStatus;
 
    virtual ECanCreateModelNPUStatus CanCreateModelNPU(const TObjectPtr<UNNEModelData> ModelData) const = 0;
    
    virtual TSharedPtr<UE::NNE::IModelNPU> CreateModelNPU(const TObjectPtr<UNNEModelData> ModelData) = 0;
};