def testTags(self): self.assertEquals(scope.CurrentDeviceScope(), None) extra_info1 = ["key1:value1"] extra_info2 = ["key2:value2"] extra_info3 = ["key3:value3"] extra_info_1_2 = ["key1:value1", "key2:value2"] extra_info_1_2_3 = ["key1:value1", "key2:value2", "key3:value3"] with scope.DeviceScope(core.DeviceOption(0, extra_info=extra_info1)): self.assertEquals(scope.CurrentDeviceScope().extra_info, extra_info1) with scope.DeviceScope(core.DeviceOption(0, extra_info=extra_info2)): self.assertEquals(scope.CurrentDeviceScope().extra_info, extra_info_1_2) with scope.DeviceScope( core.DeviceOption(0, extra_info=extra_info3)): self.assertEquals(scope.CurrentDeviceScope().extra_info, extra_info_1_2_3) self.assertEquals(scope.CurrentDeviceScope().extra_info, extra_info_1_2) self.assertEquals(scope.CurrentDeviceScope().extra_info, extra_info1) self.assertEquals(scope.CurrentDeviceScope(), None)
def testDevicescopeBasic(self): self.assertEquals(scope.CurrentDeviceScope(), None) dsc = core.DeviceOption(caffe2_pb2.CUDA, 9) with scope.DeviceScope(dsc): self.assertEquals(scope.CurrentDeviceScope(), dsc) self.assertEquals(scope.CurrentDeviceScope(), None)
def testDevicescopeBasic(self): self.assertEquals(scope.CurrentDeviceScope(), None) dsc = core.DeviceOption(_gpu_device_type(), 9) with scope.DeviceScope(dsc): self.assertEquals(scope.CurrentDeviceScope(), dsc) self.assertEquals(scope.CurrentDeviceScope(), None)
def testDevicescopeBasic(self): self.assertEquals(scope.CurrentDeviceScope(), None) dsc = core.DeviceOption(workspace.GpuDeviceType, 9) with scope.DeviceScope(dsc): self.assertEquals(scope.CurrentDeviceScope(), dsc) self.assertEquals(scope.CurrentDeviceScope(), None)
def testDevicescopeAssertion(self): self.assertEquals(scope.CurrentDeviceScope(), None) dsc = core.DeviceOption(caffe2_pb2.CUDA, 9) try: with scope.DeviceScope(dsc): self.assertEquals(scope.CurrentDeviceScope(), dsc) raise Exception() except Exception: pass self.assertEquals(scope.CurrentDeviceScope(), None)
def testDevicescopeAssertion(self): self.assertEquals(scope.CurrentDeviceScope(), None) dsc = core.DeviceOption(_gpu_device_type(), 9) try: with scope.DeviceScope(dsc): self.assertEquals(scope.CurrentDeviceScope(), dsc) raise Exception() except Exception: pass self.assertEquals(scope.CurrentDeviceScope(), None)
def testDevicescopeAssertion(self): self.assertEquals(scope.CurrentDeviceScope(), None) dsc = core.DeviceOption(workspace.GpuDeviceType, 9) try: with scope.DeviceScope(dsc): self.assertEquals(scope.CurrentDeviceScope(), dsc) raise Exception() except Exception: pass self.assertEquals(scope.CurrentDeviceScope(), None)
def thread_runner(idx, testobj): global SUCCESS_COUNT testobj.assertEquals(scope.CurrentNameScope(), "") testobj.assertEquals(scope.CurrentDeviceScope(), None) namescope = "namescope_{}".format(idx) dsc = core.DeviceOption(caffe2_pb2.CUDA, idx) with scope.DeviceScope(dsc): with scope.NameScope(namescope): testobj.assertEquals(scope.CurrentNameScope(), namescope + "/") testobj.assertEquals(scope.CurrentDeviceScope(), dsc) time.sleep(0.01 + idx * 0.01) testobj.assertEquals(scope.CurrentNameScope(), namescope + "/") testobj.assertEquals(scope.CurrentDeviceScope(), dsc) testobj.assertEquals(scope.CurrentNameScope(), "") testobj.assertEquals(scope.CurrentDeviceScope(), None) SUCCESS_COUNT += 1