Ejemplo n.º 1
0
 def gen_policy(self):
     self.policy = make_job_resource_policy(
         self.sku,
         1,  # 1 GPU per worker
         self.config,
         self.quota,
         self.metadata)
Ejemplo n.º 2
0
 def test_cpu_job(self):
     policy = make_job_resource_policy("Standard_ND24rs", 0, self.config,
                                       self.quota, self.metadata)
     self.assertIsNotNone(policy)
     self.assertEqual("1000m", policy.default_cpu_request)
     self.assertEqual("24000m", policy.default_cpu_limit)
     self.assertEqual("0Mi", policy.default_memory_request)
     self.assertEqual("458752Mi", policy.default_memory_limit)
Ejemplo n.º 3
0
 def test_gpu_job(self):
     policy = make_job_resource_policy("Standard_ND24rs", 3, self.config,
                                       self.quota, self.metadata)
     self.assertIsNotNone(policy)
     self.assertEqual("16000m", policy.default_cpu_request)
     self.assertEqual("18000m", policy.default_cpu_limit)
     self.assertEqual("309657Mi", policy.default_memory_request)
     self.assertEqual("344064Mi", policy.default_memory_limit)
Ejemplo n.º 4
0
 def test_override(self):
     self.config.update({
         "default_cpurequest": "12000m",
         "default_cpulimit": "14000m",
         "default_memoryrequest": "102400Mi",
         "default_memorylimit": "409600Mi",
     })
     policy = make_job_resource_policy("Standard_ND24rs", 0, self.config,
                                       self.quota, self.metadata)
     self.assertIsNotNone(policy)
     self.assertEqual("12000m", policy.default_cpu_request)
     self.assertEqual("14000m", policy.default_cpu_limit)
     self.assertEqual("102400Mi", policy.default_memory_request)
     self.assertEqual("409600Mi", policy.default_memory_limit)
Ejemplo n.º 5
0
 def gen_policy(self):
     self.policy = make_job_resource_policy(self.sku, self.gpu_limit,
                                            self.config, self.quota,
                                            self.metadata)