From 59ada7f406bbc71699b918679e08a787925522aa Mon Sep 17 00:00:00 2001 From: mmakevic Date: Thu, 20 Jun 2024 13:06:34 +0000 Subject: [PATCH] Fix dot_bf16 using feature conditions --- xla/lit.cfg.py | 5 +++++ xla/service/gpu/tests/dot_bf16.hlo | 5 +++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/xla/lit.cfg.py b/xla/lit.cfg.py index ebb8bd434f382..875d4ee49e125 100644 --- a/xla/lit.cfg.py +++ b/xla/lit.cfg.py @@ -43,8 +43,13 @@ ("%PYTHON", os.getenv("PYTHON", sys.executable)), ]) +if lit_config.params.get('PTX', 'GCN'): + config.available_features.add("IS_ROCM") + + # Include additional substitutions that may be defined via params config.substitutions.extend( ("%%{%s}" % key, val) for key, val in lit_config.params.items() ) + diff --git a/xla/service/gpu/tests/dot_bf16.hlo b/xla/service/gpu/tests/dot_bf16.hlo index a2a2c34c37eb8..ad3792ccd27ba 100644 --- a/xla/service/gpu/tests/dot_bf16.hlo +++ b/xla/service/gpu/tests/dot_bf16.hlo @@ -1,5 +1,6 @@ -// RUN: hlo-opt %s --platform=gpu --stage=hlo --xla_gpu_target_config_filename=%S/../../../tools/hlo_opt/gpu_specs/v100.txtpb --split-input-file | FileCheck %s --check-prefixes=CHECK-SM70 -// RUN: hlo-opt %s --platform=gpu --stage=hlo --xla_gpu_target_config_filename=%S/../../../tools/hlo_opt/gpu_specs/a100_80.txtpb --split-input-file --xla_gpu_autotune_level=0 --xla_gpu_enable_triton_gemm=false | FileCheck %s --check-prefixes=CHECK-SM80 +// RUN: %if !IS_ROCM %{ hlo-opt %s --platform=gpu --stage=hlo --xla_gpu_target_config_filename=%S/../../../tools/hlo_opt/gpu_specs/v100.txtpb --split-input-file | FileCheck %s --check-prefixes=CHECK-SM70 %} +// RUN: %if !IS_ROCM %{ hlo-opt %s --platform=gpu --stage=hlo --xla_gpu_target_config_filename=%S/../../../tools/hlo_opt/gpu_specs/a100_80.txtpb --split-input-file --xla_gpu_autotune_level=0 --xla_gpu_enable_triton_gemm=false | FileCheck %s --check-prefixes=CHECK-SM80 %} +// RUN: %if IS_ROCM %{ hlo-opt %s --platform=gpu --stage=hlo --xla_gpu_target_config_filename=%S/../../../tools/hlo_opt/gpu_specs/mi200.txtpb --split-input-file --xla_gpu_autotune_level=0 --xla_gpu_enable_triton_gemm=false | FileCheck %s --check-prefixes=CHECK-SM80 %} // CHECK-SM70: custom-call(f32