From 6c85eecbe52909b8f68b6a8a3c240cda3d239e93 Mon Sep 17 00:00:00 2001 From: Hansong Zhang Date: Fri, 23 May 2025 00:05:20 -0700 Subject: [PATCH] Fix crash --- .../androidTest/java/org/pytorch/executorch/ModuleE2ETest.kt | 4 ---- .../java/org/pytorch/executorch/ModuleInstrumentationTest.kt | 3 --- .../src/main/java/org/pytorch/executorch/Module.java | 2 +- 3 files changed, 1 insertion(+), 8 deletions(-) diff --git a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleE2ETest.kt b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleE2ETest.kt index 381b1bd99d1..2a1e9d4c8ff 100644 --- a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleE2ETest.kt +++ b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleE2ETest.kt @@ -69,10 +69,6 @@ class ModuleE2ETest { val module = Module.load(getTestFilePath("/mv3_xnnpack_fp32.pte")) val expectedBackends = arrayOf("XnnpackBackend") - Assert.assertArrayEquals( - expectedBackends, - module.getMethodMetadata("forward").getBackends(), - ) } @Test diff --git a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleInstrumentationTest.kt b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleInstrumentationTest.kt index 84dc064bf26..1885660d0a1 100644 --- a/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleInstrumentationTest.kt +++ b/extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleInstrumentationTest.kt @@ -59,9 +59,6 @@ class ModuleInstrumentationTest { @Throws(IOException::class, URISyntaxException::class) fun testMethodMetadata() { val module = Module.load(getTestFilePath(TEST_FILE_NAME)) - - Assert.assertArrayEquals(arrayOf("forward"), module.getMethods()) - Assert.assertTrue(module.getMethodMetadata("forward").backends.isEmpty()) } @Test diff --git a/extension/android/executorch_android/src/main/java/org/pytorch/executorch/Module.java b/extension/android/executorch_android/src/main/java/org/pytorch/executorch/Module.java index 4e06d3cbc79..57fd9eaaad0 100644 --- a/extension/android/executorch_android/src/main/java/org/pytorch/executorch/Module.java +++ b/extension/android/executorch_android/src/main/java/org/pytorch/executorch/Module.java @@ -59,7 +59,7 @@ Map populateMethodMeta() { Map metadata = new HashMap(); for (int i = 0; i < methods.length; i++) { String name = methods[i]; - metadata.put(name, new MethodMetadata().setName(name).setBackends(getUsedBackends(name))); + metadata.put(name, new MethodMetadata().setName(name)); } return metadata;