From bc143afd9f6cd14359efa20ef9f5721b41900515 Mon Sep 17 00:00:00 2001 From: mmakevic Date: Thu, 20 Jun 2024 13:46:38 +0000 Subject: [PATCH] Fix if statement in lit.cfg.py --- xla/lit.cfg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/xla/lit.cfg.py b/xla/lit.cfg.py index 875d4ee49e125..d148814cdfc73 100644 --- a/xla/lit.cfg.py +++ b/xla/lit.cfg.py @@ -43,7 +43,7 @@ ("%PYTHON", os.getenv("PYTHON", sys.executable)), ]) -if lit_config.params.get('PTX', 'GCN'): +if lit_config.params.get('PTX') == 'GCN': config.available_features.add("IS_ROCM")