diff --git a/requirements.txt b/requirements.txt index 59fb35c19..a3f99d862 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,5 +7,5 @@ -r pytorch-cpu-requirements.txt -r torchvision-requirements.txt -iree-compiler==20231121.715 -iree-runtime==20231121.715 +iree-compiler==20231130.724 +iree-runtime==20231130.724 diff --git a/tests/aot/globals_test.py b/tests/aot/globals_test.py index fa343b36d..6515522e1 100644 --- a/tests/aot/globals_test.py +++ b/tests/aot/globals_test.py @@ -6,6 +6,7 @@ import logging import unittest +import pytest from iree.compiler.ir import ( Context, @@ -348,6 +349,7 @@ def run(self, x=AbstractTensor(128, 20)): module_str, ) + @pytest.mark.xfail(reason="Might be IREE regression since 20231121.715") def testUninitializedParameters(self): m = SimpleParams() diff --git a/tests/dynamo/llama_test.py b/tests/dynamo/llama_test.py index 59c8e29f5..657502771 100644 --- a/tests/dynamo/llama_test.py +++ b/tests/dynamo/llama_test.py @@ -12,6 +12,7 @@ import math import unittest +import pytest from dataclasses import dataclass from typing import Any, Optional, Tuple @@ -314,6 +315,7 @@ def main(): opt(example_tokens, start_pos) +@pytest.mark.xfail(reason="https://github.com/nod-ai/SHARK-Turbine/issues/221") class ModelTests(unittest.TestCase): def testLLama(self): main() diff --git a/tests/importers/onnx_importer/import_smoke_test.py b/tests/importers/onnx_importer/import_smoke_test.py index c45a24227..e33e27120 100644 --- a/tests/importers/onnx_importer/import_smoke_test.py +++ b/tests/importers/onnx_importer/import_smoke_test.py @@ -40,16 +40,6 @@ "node_test_ai_onnx_ml_label_encoder_string_int_no_default_model", "node_test_ai_onnx_ml_label_encoder_tensor_mapping_model", "node_test_ai_onnx_ml_label_encoder_tensor_value_only_mapping_model", - "node_test_bitshift_left_uint16_model", - "node_test_bitshift_left_uint32_model", - "node_test_bitshift_left_uint64_model", - "node_test_bitshift_right_uint16_model", - "node_test_bitshift_right_uint32_model", - "node_test_bitshift_right_uint64_model", - "node_test_bitwise_and_ui64_bcast_3v1d_model", - "node_test_bitwise_not_3d_model", - "node_test_bitwise_or_ui64_bcast_3v1d_model", - "node_test_bitwise_xor_ui64_bcast_3v1d_model", "node_test_cast_FLOAT16_to_FLOAT8E4M3FNUZ_model", "node_test_cast_FLOAT16_to_FLOAT8E4M3FN_model", "node_test_cast_FLOAT16_to_FLOAT8E5M2FNUZ_model", @@ -166,15 +156,6 @@ "node_test_lstm_defaults_model", "node_test_lstm_with_initial_bias_model", "node_test_lstm_with_peepholes_model", - "node_test_max_uint16_model", - "node_test_max_uint32_model", - "node_test_max_uint64_model", - "node_test_min_uint16_model", - "node_test_min_uint32_model", - "node_test_min_uint64_model", - "node_test_mod_uint16_model", - "node_test_mod_uint32_model", - "node_test_mod_uint64_model", "node_test_optional_get_element_optional_sequence_model", "node_test_optional_get_element_optional_tensor_model", "node_test_optional_get_element_sequence_model", @@ -183,8 +164,6 @@ "node_test_optional_has_element_empty_optional_input_model", "node_test_optional_has_element_optional_input_model", "node_test_optional_has_element_tensor_input_model", - "node_test_pow_types_float32_uint32_model", - "node_test_pow_types_float32_uint64_model", "node_test_quantizelinear_e4m3fn_model", "node_test_quantizelinear_e5m2_model", "node_test_range_float_type_positive_delta_expanded_model",