Fix if statement in lit.cfg.py

ROCm · Jun 20, 2024 · bc143af · bc143af
1 parent 59ada7f
commit bc143af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")