Python _TpuGroupSpecの例

コード例 #1

 def testTpuZone(self):
     FLAGS['tpu_zone'].parse('us-central1-c')
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.full_spec)
     self.assertEqual(result.tpu_zone, 'us-central1-c')

コード例 #2

 def testCustomTpuName(self):
     spec = MergeDicts(self.minimal_spec, {'tpu_name': 'pkb-tpu'})
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **spec)
     self.assertEqual(result.tpu_name, 'pkb-tpu')

コード例 #3

 def testTpuCidrRangeFlag(self):
     FLAGS['tpu_cidr_range'].parse('10.240.0.0/29')
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.full_spec)
     self.assertEqual(result.tpu_cidr_range, '10.240.0.0/29')

コード例 #4

 def testTpuAcceleratorTypeFlag(self):
     FLAGS['tpu_accelerator_type'].parse('tpu-v1')
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.full_spec)
     self.assertEqual(result.tpu_accelerator_type, 'tpu-v1')

コード例 #5

 def testCustomTpuVersion(self):
     spec = MergeDicts(self.minimal_spec, {'tpu_tf_version': 'nightly'})
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **spec)
     self.assertEqual(result.tpu_tf_version, 'nightly')

コード例 #6

 def testTpuNameFlag(self):
     FLAGS['tpu_name'].parse('pkb-tpu')
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.full_spec)
     self.assertEqual(result.tpu_name, 'pkb-tpu')

コード例 #7

 def testCustomTpuZone(self):
     spec = MergeDicts(self.minimal_spec, {'tpu_zone': 'us-central1-a'})
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **spec)
     self.assertEqual(result.tpu_zone, 'us-central1-a')

コード例 #8

 def testTpuNetworkFlag(self):
     FLAGS['tpu_network'].parse('my-tf-network')
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.full_spec)
     self.assertEqual(result.tpu_network, 'my-tf-network')

コード例 #9

 def testCustomTpuNetwork(self):
     spec = MergeDicts(self.minimal_spec, {'tpu_network': 'default'})
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **spec)
     self.assertEqual(result.tpu_network, 'default')

コード例 #10

 def testCustomTpuDescription(self):
     spec = MergeDicts(self.minimal_spec, {'tpu_description': 'My TF Node'})
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **spec)
     self.assertEqual(result.tpu_description, 'My TF Node')

コード例 #11

 def testTpuDescriptionFlag(self):
     FLAGS['tpu_description'].parse('MyTfNode')
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.full_spec)
     self.assertEqual(result.tpu_description, 'MyTfNode')

コード例 #12

 def testTpuTfVersion(self):
     FLAGS['tpu_tf_version'].parse('1.2')
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.full_spec)
     self.assertEqual(result.tpu_tf_version, '1.2')

コード例 #13

 def testCustomTpuAcceleratorType(self):
     spec = MergeDicts(self.minimal_spec,
                       {'tpu_accelerator_type': 'tpu-v2'})
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **spec)
     self.assertEqual(result.tpu_accelerator_type, 'tpu-v2')

コード例 #14

 def testCustomTpuCidrRange(self):
     spec = MergeDicts(self.minimal_spec,
                       {'tpu_cidr_range': '192.168.0.0/29'})
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **spec)
     self.assertEqual(result.tpu_cidr_range, '192.168.0.0/29')

コード例 #15

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testTpuNameFlag(self):
   self.flags['tpu_name'].parse('pkb-tpu')
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.full_spec)
   self.assertEqual(result.tpu_name, 'pkb-tpu')

コード例 #16

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testDefaultTpuVersion(self):
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.minimal_spec)
   self.assertEqual(result.tpu_tf_version, None)

コード例 #17

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testCustomTpuVersion(self):
   spec = MergeDicts(self.minimal_spec, {'tpu_tf_version': 'nightly'})
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **spec)
   self.assertEqual(result.tpu_tf_version, 'nightly')

コード例 #18

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testCustomTpuNetwork(self):
   spec = MergeDicts(self.minimal_spec, {'tpu_network': 'default'})
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **spec)
   self.assertEqual(result.tpu_network, 'default')

コード例 #19

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testCustomTpuZone(self):
   spec = MergeDicts(self.minimal_spec, {'tpu_zone': 'us-central1-a'})
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **spec)
   self.assertEqual(result.tpu_zone, 'us-central1-a')

コード例 #20

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testCustomTpuAcceleratorType(self):
   spec = MergeDicts(self.minimal_spec, {'tpu_accelerator_type': 'tpu-v2'})
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **spec)
   self.assertEqual(result.tpu_accelerator_type, 'tpu-v2')

コード例 #21

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testCustomTpuDescription(self):
   spec = MergeDicts(self.minimal_spec, {'tpu_description': 'My TF Node'})
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **spec)
   self.assertEqual(result.tpu_description, 'My TF Node')

コード例 #22

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testCustomTpuCidrRange(self):
   spec = MergeDicts(self.minimal_spec, {'tpu_cidr_range': '192.168.0.0/29'})
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **spec)
   self.assertEqual(result.tpu_cidr_range, '192.168.0.0/29')

コード例 #23

ファイル: cloud_tpu_test.py プロジェクト: xiaoyuanW/PerfKitBenchmarker

 def testDefaultTpuZone(self):
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=self.flags,
                                                  **self.minimal_spec)
     self.assertEqual(result.tpu_zone, None)

コード例 #24

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testCustomTpuName(self):
   spec = MergeDicts(self.minimal_spec, {'tpu_name': 'pkb-tpu'})
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **spec)
   self.assertEqual(result.tpu_name, 'pkb-tpu')

コード例 #25

 def testDefaultTpuAcceleratorType(self):
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.minimal_spec)
     self.assertEqual(result.tpu_accelerator_type, None)

コード例 #26

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testTpuDescriptionFlag(self):
   self.flags['tpu_description'].parse('MyTfNode')
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.full_spec)
   self.assertEqual(result.tpu_description, 'MyTfNode')

コード例 #27

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testTpuCidrRangeFlag(self):
   self.flags['tpu_cidr_range'].parse('10.240.0.0/29')
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.full_spec)
   self.assertEqual(result.tpu_cidr_range, '10.240.0.0/29')

コード例 #28

 def testDefaultTpuVersion(self):
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.minimal_spec)
     self.assertEqual(result.tpu_tf_version, None)

コード例 #29

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testTpuAcceleratorTypeFlag(self):
   self.flags['tpu_accelerator_type'].parse('tpu-v1')
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.full_spec)
   self.assertEqual(result.tpu_accelerator_type, 'tpu-v1')

コード例 #30

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testTpuTfVersion(self):
   self.flags['tpu_tf_version'].parse('1.2')
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.full_spec)
   self.assertEqual(result.tpu_tf_version, '1.2')

コード例 #31

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testTpuNetworkFlag(self):
   self.flags['tpu_network'].parse('my-tf-network')
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.full_spec)
   self.assertEqual(result.tpu_network, 'my-tf-network')

コード例 #32

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testMinimalConfig(self):
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.minimal_spec)
   self.assertEqual(result.cloud, 'GCP')

コード例 #33

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testTpuZone(self):
   self.flags['tpu_zone'].parse('us-central1-c')
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.full_spec)
   self.assertEqual(result.tpu_zone, 'us-central1-c')

コード例 #34

 def testMinimalConfig(self):
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.minimal_spec)
     self.assertEqual(result.cloud, 'GCP')

コード例 #35

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testDefaultTpuName(self):
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.minimal_spec)
   self.assertEqual(result.tpu_name, 'pkb-tpu-train-123')

コード例 #36

 def testDefaultTpuName(self):
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.minimal_spec)
     self.assertEqual(result.tpu_name, 'pkb-tpu-train-123')

コード例 #37

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testDefaultTpuCidrRange(self):
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.minimal_spec)
   self.assertEqual(result.tpu_cidr_range, None)

コード例 #38

 def testDefaultTpuCidrRange(self):
     result = benchmark_config_spec._TpuGroupSpec(_COMPONENT,
                                                  _GROUP_NAME,
                                                  flag_values=FLAGS,
                                                  **self.minimal_spec)
     self.assertEqual(result.tpu_cidr_range, None)

コード例 #39

ファイル: cloud_tpu_test.py プロジェクト: 9723/PerfKitBenchmarker

 def testDefaultTpuAcceleratorType(self):
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=self.flags, **self.minimal_spec)
   self.assertEqual(result.tpu_accelerator_type, None)

コード例 #40

ファイル: cloud_tpu_test.py プロジェクト: GoogleCloudPlatform/PerfKitBenchmarker

 def testDefaultTpuNetwork(self):
   result = benchmark_config_spec._TpuGroupSpec(
       _COMPONENT, _GROUP_NAME, flag_values=FLAGS, **self.minimal_spec)
   self.assertEqual(result.tpu_network, None)