ComputeQuotaResourceConfig.h

 
#pragma once
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/AcceleratorPartitionConfig.h>
#include <aws/sagemaker/model/ClusterInstanceType.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class ComputeQuotaResourceConfig {
 public:
  AWS_SAGEMAKER_API ComputeQuotaResourceConfig() = default;
  AWS_SAGEMAKER_API ComputeQuotaResourceConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API ComputeQuotaResourceConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline ClusterInstanceType GetInstanceType() const { return m_instanceType; }
  inline bool InstanceTypeHasBeenSet() const { return m_instanceTypeHasBeenSet; }
  inline void SetInstanceType(ClusterInstanceType value) {
    m_instanceTypeHasBeenSet = true;
    m_instanceType = value;
  }
  inline ComputeQuotaResourceConfig& WithInstanceType(ClusterInstanceType value) {
    SetInstanceType(value);
    return *this;
  }
 
 
  inline int GetCount() const { return m_count; }
  inline bool CountHasBeenSet() const { return m_countHasBeenSet; }
  inline void SetCount(int value) {
    m_countHasBeenSet = true;
    m_count = value;
  }
  inline ComputeQuotaResourceConfig& WithCount(int value) {
    SetCount(value);
    return *this;
  }
 
 
  inline int GetAccelerators() const { return m_accelerators; }
  inline bool AcceleratorsHasBeenSet() const { return m_acceleratorsHasBeenSet; }
  inline void SetAccelerators(int value) {
    m_acceleratorsHasBeenSet = true;
    m_accelerators = value;
  }
  inline ComputeQuotaResourceConfig& WithAccelerators(int value) {
    SetAccelerators(value);
    return *this;
  }
 
 
  inline double GetVCpu() const { return m_vCpu; }
  inline bool VCpuHasBeenSet() const { return m_vCpuHasBeenSet; }
  inline void SetVCpu(double value) {
    m_vCpuHasBeenSet = true;
    m_vCpu = value;
  }
  inline ComputeQuotaResourceConfig& WithVCpu(double value) {
    SetVCpu(value);
    return *this;
  }
 
 
  inline double GetMemoryInGiB() const { return m_memoryInGiB; }
  inline bool MemoryInGiBHasBeenSet() const { return m_memoryInGiBHasBeenSet; }
  inline void SetMemoryInGiB(double value) {
    m_memoryInGiBHasBeenSet = true;
    m_memoryInGiB = value;
  }
  inline ComputeQuotaResourceConfig& WithMemoryInGiB(double value) {
    SetMemoryInGiB(value);
    return *this;
  }
 
 
  inline const AcceleratorPartitionConfig& GetAcceleratorPartition() const { return m_acceleratorPartition; }
  inline bool AcceleratorPartitionHasBeenSet() const { return m_acceleratorPartitionHasBeenSet; }
  template <typename AcceleratorPartitionT = AcceleratorPartitionConfig>
  void SetAcceleratorPartition(AcceleratorPartitionT&& value) {
    m_acceleratorPartitionHasBeenSet = true;
    m_acceleratorPartition = std::forward<AcceleratorPartitionT>(value);
  }
  template <typename AcceleratorPartitionT = AcceleratorPartitionConfig>
  ComputeQuotaResourceConfig& WithAcceleratorPartition(AcceleratorPartitionT&& value) {
    SetAcceleratorPartition(std::forward<AcceleratorPartitionT>(value));
    return *this;
  }
 private:
  ClusterInstanceType m_instanceType{ClusterInstanceType::NOT_SET};
 
  int m_count{0};
 
  int m_accelerators{0};
 
  double m_vCpu{0.0};
 
  double m_memoryInGiB{0.0};
 
  AcceleratorPartitionConfig m_acceleratorPartition;
  bool m_instanceTypeHasBeenSet = false;
  bool m_countHasBeenSet = false;
  bool m_acceleratorsHasBeenSet = false;
  bool m_vCpuHasBeenSet = false;
  bool m_memoryInGiBHasBeenSet = false;
  bool m_acceleratorPartitionHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws