From 7835c9042749dbdc8934850720944c876d6c37be Mon Sep 17 00:00:00 2001 From: stanley-nod Date: Thu, 21 Sep 2023 10:56:22 -0700 Subject: [PATCH] Add a xfail test for scalars on args_test --- tests/aot/args_test.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/aot/args_test.py b/tests/aot/args_test.py index f833b888e..fd9bf526b 100644 --- a/tests/aot/args_test.py +++ b/tests/aot/args_test.py @@ -77,6 +77,22 @@ def compute(a, b): module_str, ) + @unittest.expectedFailure + def testProcToJitScalarArgs(self): + """Expected to fail, missing jittable support for IrValueScalar.""" + + class BasicModule(CompiledModule): + def foobar(self, a=AbstractI32, b=AbstractI32): + return self.compute(a, b) + + @jittable + def compute(a, b): + return a + b + + inst = BasicModule(context=Context()) + module_str = str(CompiledModule.get_mlir_module(inst)) + print(module_str) + if __name__ == "__main__": logging.basicConfig(level=logging.DEBUG)