"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "1afc21855eb1f5575bd61037a7ee44522ccf401e"
Commit 92af450b authored by generatedunixname2443911735787003's avatar generatedunixname2443911735787003 Committed by Facebook GitHub Bot
Browse files

fbcode//mobile-vision/d2go/tests

Summary: Pull Request resolved: https://github.com/facebookresearch/d2go/pull/648

Differential Revision: D52792992

fbshipit-source-id: d3a64f3b306ea024ec072eaa6327446a84e2d83c
parent 573bd454
...@@ -34,24 +34,24 @@ class TestTorchscriptExportMethods(unittest.TestCase): ...@@ -34,24 +34,24 @@ class TestTorchscriptExportMethods(unittest.TestCase):
export_method_string = "torchscript_mobile" export_method_string = "torchscript_mobile"
new_export_kwargs = try_mock_export(export_method_string) new_export_kwargs = try_mock_export(export_method_string)
self.assertIn("mobile_optimization", new_export_kwargs) self.assertIn("mobile_optimization", new_export_kwargs)
self.assertEquals( self.assertEqual(
type(new_export_kwargs["mobile_optimization"]), type(new_export_kwargs["mobile_optimization"]),
MobileOptimizationConfig, MobileOptimizationConfig,
) )
self.assertEquals(new_export_kwargs["mobile_optimization"].backend, "CPU") self.assertEqual(new_export_kwargs["mobile_optimization"].backend, "CPU")
export_method_string = "torchscript_mobile-metal" export_method_string = "torchscript_mobile-metal"
new_export_kwargs = try_mock_export(export_method_string) new_export_kwargs = try_mock_export(export_method_string)
self.assertEquals(new_export_kwargs["mobile_optimization"].backend, "metal") self.assertEqual(new_export_kwargs["mobile_optimization"].backend, "metal")
export_method_string = "torchscript_mobile-vulkan" export_method_string = "torchscript_mobile-vulkan"
new_export_kwargs = try_mock_export(export_method_string) new_export_kwargs = try_mock_export(export_method_string)
self.assertEquals(new_export_kwargs["mobile_optimization"].backend, "vulkan") self.assertEqual(new_export_kwargs["mobile_optimization"].backend, "vulkan")
export_method_string = "torchscript_mobile@tracing" export_method_string = "torchscript_mobile@tracing"
new_export_kwargs = try_mock_export(export_method_string) new_export_kwargs = try_mock_export(export_method_string)
self.assertEquals(new_export_kwargs["jit_mode"], "trace") self.assertEqual(new_export_kwargs["jit_mode"], "trace")
export_method_string = "torchscript_mobile@scripting" export_method_string = "torchscript_mobile@scripting"
new_export_kwargs = try_mock_export(export_method_string) new_export_kwargs = try_mock_export(export_method_string)
self.assertEquals(new_export_kwargs["jit_mode"], "script") self.assertEqual(new_export_kwargs["jit_mode"], "script")
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment