Skip to content

FGPURuntimeWrapper

Wrapper class for GPU-based neural network runtimes.

This class wraps NNE GPU runtimes to provide a consistent interface for creating models and checking validity.

class FGPURuntimeWrapper : public IRuntimeInterface

Methods

FGPURuntimeWrapper

inline FGPURuntimeWrapper(TWeakInterfacePtr<INNERuntimeGPU> RawRuntime)

Constructor that wraps a raw GPU runtime.

Parameters

  • RawRuntime (TWeakInterfacePtr<INNERuntimeGPU>)

IsValid

inline virtual bool IsValid()

Checks if the wrapped runtime is valid.


CreateModel

inline virtual TUniquePtr<IModelInterface>
CreateModel(TObjectPtr<UNNEModelData> ModelData)

Creates a GPU model from model data.

Parameters

  • ModelData (TObjectPtr<UNNEModelData>)

Source: Source/ScholaNNE/Public/NNEUtils/NNEWrappers.h