InferenceComponentComputeResourceRequirements.h

 
#pragma once
#include <aws/sagemaker/SageMaker_EXPORTS.h>
 
namespace Aws
{
namespace Utils
{
namespace Json
{
  class JsonValue;
  class JsonView;
} // namespace Json
} // namespace Utils
namespace SageMaker
{
namespace Model
{
 
  class InferenceComponentComputeResourceRequirements
  {
  public:
    AWS_SAGEMAKER_API InferenceComponentComputeResourceRequirements();
    AWS_SAGEMAKER_API InferenceComponentComputeResourceRequirements(Aws::Utils::Json::JsonView jsonValue);
    AWS_SAGEMAKER_API InferenceComponentComputeResourceRequirements& operator=(Aws::Utils::Json::JsonView jsonValue);
    AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
 
    inline double GetNumberOfCpuCoresRequired() const{ return m_numberOfCpuCoresRequired; }
    inline bool NumberOfCpuCoresRequiredHasBeenSet() const { return m_numberOfCpuCoresRequiredHasBeenSet; }
    inline void SetNumberOfCpuCoresRequired(double value) { m_numberOfCpuCoresRequiredHasBeenSet = true; m_numberOfCpuCoresRequired = value; }
    inline InferenceComponentComputeResourceRequirements& WithNumberOfCpuCoresRequired(double value) { SetNumberOfCpuCoresRequired(value); return *this;}
 
 
    inline double GetNumberOfAcceleratorDevicesRequired() const{ return m_numberOfAcceleratorDevicesRequired; }
    inline bool NumberOfAcceleratorDevicesRequiredHasBeenSet() const { return m_numberOfAcceleratorDevicesRequiredHasBeenSet; }
    inline void SetNumberOfAcceleratorDevicesRequired(double value) { m_numberOfAcceleratorDevicesRequiredHasBeenSet = true; m_numberOfAcceleratorDevicesRequired = value; }
    inline InferenceComponentComputeResourceRequirements& WithNumberOfAcceleratorDevicesRequired(double value) { SetNumberOfAcceleratorDevicesRequired(value); return *this;}
 
 
    inline int GetMinMemoryRequiredInMb() const{ return m_minMemoryRequiredInMb; }
    inline bool MinMemoryRequiredInMbHasBeenSet() const { return m_minMemoryRequiredInMbHasBeenSet; }
    inline void SetMinMemoryRequiredInMb(int value) { m_minMemoryRequiredInMbHasBeenSet = true; m_minMemoryRequiredInMb = value; }
    inline InferenceComponentComputeResourceRequirements& WithMinMemoryRequiredInMb(int value) { SetMinMemoryRequiredInMb(value); return *this;}
 
 
    inline int GetMaxMemoryRequiredInMb() const{ return m_maxMemoryRequiredInMb; }
    inline bool MaxMemoryRequiredInMbHasBeenSet() const { return m_maxMemoryRequiredInMbHasBeenSet; }
    inline void SetMaxMemoryRequiredInMb(int value) { m_maxMemoryRequiredInMbHasBeenSet = true; m_maxMemoryRequiredInMb = value; }
    inline InferenceComponentComputeResourceRequirements& WithMaxMemoryRequiredInMb(int value) { SetMaxMemoryRequiredInMb(value); return *this;}
  private:
 
    double m_numberOfCpuCoresRequired;
    bool m_numberOfCpuCoresRequiredHasBeenSet = false;
 
    double m_numberOfAcceleratorDevicesRequired;
    bool m_numberOfAcceleratorDevicesRequiredHasBeenSet = false;
 
    int m_minMemoryRequiredInMb;
    bool m_minMemoryRequiredInMbHasBeenSet = false;
 
    int m_maxMemoryRequiredInMb;
    bool m_maxMemoryRequiredInMbHasBeenSet = false;
  };
 
} // namespace Model
} // namespace SageMaker
} // namespace Aws