github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-11 06:48:09 -05:00
Code Issues Packages Projects Releases Wiki Activity

Test tflite quantization_info fail on cloud (#228)

Browse Source
This commit is contained in:
Chi_Liu
2022-08-01 18:21:04 -07:00
committed by GitHub
parent 0c31bb82cd
commit 0e42c19f33
4 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tank/arbitrary-image-stylization-v1-256/arbitrary-image-stylization-v1-256_tflite_test.py
View File

@@ -77,9 +77,10 @@ class ArbitraryImageStylizationV1TfliteModuleTest(unittest.TestCase):
)
self.module_tester.save_mlir = self.save_mlir
@pytest.mark.skip(
reason="known macos tflite install issue & "
"'tosa.conv2d' op attribute 'quantization_info' failed "
import sys
@pytest.mark.xfail(
reason="'tosa.conv2d' op attribute 'quantization_info' failed ",
)
def test_module_static_cpu(self):
self.module_tester.dynamic = False

7
tank/birds_V1/birds_V1_tflite_test.py
View File

@@ -111,9 +111,10 @@ class BirdsV1TfliteModuleTest(unittest.TestCase):
self.module_tester = BirdsV1TfliteModuleTester(self)
self.module_tester.save_mlir = self.save_mlir
@pytest.mark.skip(
reason="known macos tflite install issue & "
"'tosa.conv2d' op attribute 'quantization_info' failed "
import sys
@pytest.mark.xfail(
reason="'tosa.conv2d' op attribute 'quantization_info' failed ",
)
def test_module_static_cpu(self):
self.module_tester.dynamic = False

3
tank/gpt2-64/gpt2-64_tflite_test.py
View File

@@ -32,7 +32,7 @@ def compare_results(mlir_results, tflite_results):
tflite_result = tflite_results[i]
mlir_result = mlir_result.astype(np.single)
tflite_result = tflite_result.astype(np.single)
mlir_result = np.expand_dims(mlir_result, axis=0)
# mlir_result = np.expand_dims(mlir_result, axis=0)
print("mlir_result.shape", mlir_result.shape)
print("tflite_result.shape", tflite_result.shape)
assert mlir_result.shape == tflite_result.shape, "shape doesnot match"
@@ -104,7 +104,6 @@ class GptTfliteModuleTest(unittest.TestCase):
self.module_tester = GptTfliteModuleTester(self)
self.module_tester.save_mlir = self.save_mlir
@pytest.mark.skip(reason="gpt2-64.tflite model too big")
def test_module_static_cpu(self):
self.module_tester.dynamic = False
self.module_tester.device = "cpu"

1
tank/nasnet/nasnet_tflite_test.py
View File

@@ -73,7 +73,6 @@ class NasnetTfliteModuleTest(unittest.TestCase):
self.module_tester = NasnetTfliteModuleTester(self)
self.module_tester.save_mlir = self.save_mlir
@pytest.mark.skip(reason="nasnet tflite model too big")
def test_module_static_cpu(self):
self.module_tester.dynamic = False
self.module_tester.device = "cpu"