X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/af3bfbb6126ebce381c52ced87fc8af372d84118..f84e4b2ab7cd923aff2f99c04cb1313b36866393:/lib/dispatchcloud/node_size_test.go diff --git a/lib/dispatchcloud/node_size_test.go b/lib/dispatchcloud/node_size_test.go index d6b7c6bf9f..eb3648e8ac 100644 --- a/lib/dispatchcloud/node_size_test.go +++ b/lib/dispatchcloud/node_size_test.go @@ -5,13 +5,13 @@ package dispatchcloud import ( - "git.curoverse.com/arvados.git/sdk/go/arvados" + "git.arvados.org/arvados.git/sdk/go/arvados" check "gopkg.in/check.v1" ) var _ = check.Suite(&NodeSizeSuite{}) -const GiB = int64(1 << 30) +const GiB = arvados.ByteSize(1 << 30) type NodeSizeSuite struct{} @@ -27,10 +27,10 @@ func (*NodeSizeSuite) TestChooseNotConfigured(c *check.C) { func (*NodeSizeSuite) TestChooseUnsatisfiable(c *check.C) { checkUnsatisfiable := func(ctr *arvados.Container) { - _, err := ChooseInstanceType(&arvados.Cluster{InstanceTypes: []arvados.InstanceType{ - {Price: 1.1, RAM: 1000000000, VCPUs: 2, Name: "small1"}, - {Price: 2.2, RAM: 2000000000, VCPUs: 4, Name: "small2"}, - {Price: 4.4, RAM: 4000000000, VCPUs: 8, Name: "small4", Scratch: GiB}, + _, err := ChooseInstanceType(&arvados.Cluster{InstanceTypes: map[string]arvados.InstanceType{ + "small1": {Price: 1.1, RAM: 1000000000, VCPUs: 2, Name: "small1"}, + "small2": {Price: 2.2, RAM: 2000000000, VCPUs: 4, Name: "small2"}, + "small4": {Price: 4.4, RAM: 4000000000, VCPUs: 8, Name: "small4", Scratch: GiB}, }}, ctr) c.Check(err, check.FitsTypeOf, ConstraintsNotSatisfiableError{}) } @@ -43,40 +43,46 @@ func (*NodeSizeSuite) TestChooseUnsatisfiable(c *check.C) { checkUnsatisfiable(&arvados.Container{RuntimeConstraints: rc}) } checkUnsatisfiable(&arvados.Container{ - Mounts: map[string]arvados.Mount{"/tmp": {Kind: "tmp", Capacity: 2 * GiB}}, + Mounts: map[string]arvados.Mount{"/tmp": {Kind: "tmp", Capacity: int64(2 * GiB)}}, RuntimeConstraints: arvados.RuntimeConstraints{RAM: 12345, VCPUs: 1}, }) } func (*NodeSizeSuite) TestChoose(c *check.C) { - for _, menu := range [][]arvados.InstanceType{ + for _, menu := range []map[string]arvados.InstanceType{ { - {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly"}, - {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best"}, - {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: 2 * GiB, Name: "small"}, + "costly": {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly"}, + "best": {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best"}, + "small": {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: 2 * GiB, Name: "small"}, }, { - {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly"}, - {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "goodenough"}, - {Price: 2.2, RAM: 4000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best"}, - {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: 2 * GiB, Name: "small"}, + "costly": {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly"}, + "goodenough": {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "goodenough"}, + "best": {Price: 2.2, RAM: 4000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best"}, + "small": {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: 2 * GiB, Name: "small"}, }, { - {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: 2 * GiB, Name: "small"}, - {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "goodenough"}, - {Price: 2.2, RAM: 4000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best"}, - {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly"}, + "small": {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: 2 * GiB, Name: "small"}, + "goodenough": {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "goodenough"}, + "best": {Price: 2.2, RAM: 4000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best"}, + "costly": {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly"}, }, { - {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: GiB, Name: "small"}, - {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: GiB, Name: "nearly"}, - {Price: 3.3, RAM: 4000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best"}, - {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly"}, + "small": {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: GiB, Name: "small"}, + "nearly": {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: GiB, Name: "nearly"}, + "best": {Price: 3.3, RAM: 4000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best"}, + "costly": {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly"}, + }, + { + "small": {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: GiB, Name: "small"}, + "nearly": {Price: 2.2, RAM: 1200000000, VCPUs: 4, Scratch: 2 * GiB, Name: "nearly"}, + "best": {Price: 3.3, RAM: 4000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best"}, + "costly": {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly"}, }, } { - best, err := ChooseInstanceType(&arvados.Cluster{InstanceTypes: menu}, &arvados.Container{ + best, err := ChooseInstanceType(&arvados.Cluster{InstanceTypes: menu, Containers: arvados.ContainersConfig{ReserveExtraRAM: 268435456}}, &arvados.Container{ Mounts: map[string]arvados.Mount{ - "/tmp": {Kind: "tmp", Capacity: 2 * GiB}, + "/tmp": {Kind: "tmp", Capacity: 2 * int64(GiB)}, }, RuntimeConstraints: arvados.RuntimeConstraints{ VCPUs: 2, @@ -93,15 +99,15 @@ func (*NodeSizeSuite) TestChoose(c *check.C) { } func (*NodeSizeSuite) TestChoosePreemptable(c *check.C) { - menu := []arvados.InstanceType{ - {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Preemptable: true, Name: "costly"}, - {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "almost best"}, - {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Preemptable: true, Name: "best"}, - {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: 2 * GiB, Preemptable: true, Name: "small"}, + menu := map[string]arvados.InstanceType{ + "costly": {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Preemptible: true, Name: "costly"}, + "almost best": {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "almost best"}, + "best": {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Preemptible: true, Name: "best"}, + "small": {Price: 1.1, RAM: 1000000000, VCPUs: 2, Scratch: 2 * GiB, Preemptible: true, Name: "small"}, } best, err := ChooseInstanceType(&arvados.Cluster{InstanceTypes: menu}, &arvados.Container{ Mounts: map[string]arvados.Mount{ - "/tmp": {Kind: "tmp", Capacity: 2 * GiB}, + "/tmp": {Kind: "tmp", Capacity: 2 * int64(GiB)}, }, RuntimeConstraints: arvados.RuntimeConstraints{ VCPUs: 2, @@ -109,7 +115,7 @@ func (*NodeSizeSuite) TestChoosePreemptable(c *check.C) { KeepCacheRAM: 123456789, }, SchedulingParameters: arvados.SchedulingParameters{ - Preemptable: true, + Preemptible: true, }, }) c.Check(err, check.IsNil) @@ -117,5 +123,114 @@ func (*NodeSizeSuite) TestChoosePreemptable(c *check.C) { c.Check(best.RAM >= 1234567890, check.Equals, true) c.Check(best.VCPUs >= 2, check.Equals, true) c.Check(best.Scratch >= 2*GiB, check.Equals, true) - c.Check(best.Preemptable, check.Equals, true) + c.Check(best.Preemptible, check.Equals, true) +} + +func (*NodeSizeSuite) TestScratchForDockerImage(c *check.C) { + n := EstimateScratchSpace(&arvados.Container{ + ContainerImage: "d5025c0f29f6eef304a7358afa82a822+342", + }) + // Actual image is 371.1 MiB (according to workbench) + // Estimated size is 384 MiB (402653184 bytes) + // Want to reserve 2x the estimated size, so 805306368 bytes + c.Check(n, check.Equals, int64(805306368)) + + n = EstimateScratchSpace(&arvados.Container{ + ContainerImage: "d5025c0f29f6eef304a7358afa82a822+-342", + }) + // Parse error will return 0 + c.Check(n, check.Equals, int64(0)) + + n = EstimateScratchSpace(&arvados.Container{ + ContainerImage: "d5025c0f29f6eef304a7358afa82a822+34", + }) + // Short manifest will return 0 + c.Check(n, check.Equals, int64(0)) +} + +func (*NodeSizeSuite) TestChooseGPU(c *check.C) { + menu := map[string]arvados.InstanceType{ + "costly": {Price: 4.4, RAM: 4000000000, VCPUs: 8, Scratch: 2 * GiB, Name: "costly", CUDA: arvados.CUDAFeatures{DeviceCount: 2, HardwareCapability: "9.0", DriverVersion: "11.0"}}, + "low_capability": {Price: 2.1, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "low_capability", CUDA: arvados.CUDAFeatures{DeviceCount: 1, HardwareCapability: "8.0", DriverVersion: "11.0"}}, + "best": {Price: 2.2, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "best", CUDA: arvados.CUDAFeatures{DeviceCount: 1, HardwareCapability: "9.0", DriverVersion: "11.0"}}, + "low_driver": {Price: 2.1, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "low_driver", CUDA: arvados.CUDAFeatures{DeviceCount: 1, HardwareCapability: "9.0", DriverVersion: "10.0"}}, + "cheap_gpu": {Price: 2.0, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "cheap_gpu", CUDA: arvados.CUDAFeatures{DeviceCount: 1, HardwareCapability: "8.0", DriverVersion: "10.0"}}, + "invalid_gpu": {Price: 1.9, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "invalid_gpu", CUDA: arvados.CUDAFeatures{DeviceCount: 1, HardwareCapability: "12.0.12", DriverVersion: "12.0.12"}}, + "non_gpu": {Price: 1.1, RAM: 2000000000, VCPUs: 4, Scratch: 2 * GiB, Name: "non_gpu"}, + } + + type GPUTestCase struct { + CUDA arvados.CUDARuntimeConstraints + SelectedInstance string + } + cases := []GPUTestCase{ + GPUTestCase{ + CUDA: arvados.CUDARuntimeConstraints{ + DeviceCount: 1, + HardwareCapability: "9.0", + DriverVersion: "11.0", + }, + SelectedInstance: "best", + }, + GPUTestCase{ + CUDA: arvados.CUDARuntimeConstraints{ + DeviceCount: 2, + HardwareCapability: "9.0", + DriverVersion: "11.0", + }, + SelectedInstance: "costly", + }, + GPUTestCase{ + CUDA: arvados.CUDARuntimeConstraints{ + DeviceCount: 1, + HardwareCapability: "8.0", + DriverVersion: "11.0", + }, + SelectedInstance: "low_capability", + }, + GPUTestCase{ + CUDA: arvados.CUDARuntimeConstraints{ + DeviceCount: 1, + HardwareCapability: "9.0", + DriverVersion: "10.0", + }, + SelectedInstance: "low_driver", + }, + GPUTestCase{ + CUDA: arvados.CUDARuntimeConstraints{ + DeviceCount: 1, + HardwareCapability: "", + DriverVersion: "10.0", + }, + SelectedInstance: "", + }, + GPUTestCase{ + CUDA: arvados.CUDARuntimeConstraints{ + DeviceCount: 0, + HardwareCapability: "9.0", + DriverVersion: "11.0", + }, + SelectedInstance: "non_gpu", + }, + } + + for _, tc := range cases { + best, err := ChooseInstanceType(&arvados.Cluster{InstanceTypes: menu}, &arvados.Container{ + Mounts: map[string]arvados.Mount{ + "/tmp": {Kind: "tmp", Capacity: 2 * int64(GiB)}, + }, + RuntimeConstraints: arvados.RuntimeConstraints{ + VCPUs: 2, + RAM: 987654321, + KeepCacheRAM: 123456789, + CUDA: tc.CUDA, + }, + }) + if best.Name != "" { + c.Check(err, check.IsNil) + c.Check(best.Name, check.Equals, tc.SelectedInstance) + } else { + c.Check(err, check.Not(check.IsNil)) + } + } }