BackendBase.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "DefinitionsAI.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class CComputationalGraphUtilities;
 
        template <typename T>
        class FL_EXPORT CBackendBase : public CAlgorithmAIBase
        {
 
        public:
            CBackendBase();
            virtual ~CBackendBase();
 
            virtual const CResult SetDeviceIndex(int32_t i32DeviceIndex);
            virtual int32_t GetDeviceIndex() const;
 
            virtual const CResult EnableTensorCore(bool bTensorCore = true);
            virtual bool IsTensorCoreEnabled() const;
 
            virtual const CResult Clear();
            DeclareGetClassType();
            SupportToDuplicateAbstractObject(CBackendBase<T>);
 
        protected:
            virtual void Throw(const CResult& res, const wchar_t* pWcsExtraMessage = nullptr) const override;
        protected:
            int32_t m_i32DeviceIndex;
            bool m_bTensorCoreEnabled;
 
        protected:
            template <typename T>
            friend class CComputationalGraphUtilities;
        };
    }
}
 
#endif