diff --git a/tuner/tuner/common.py b/tuner/tuner/common.py index b01d93125..bcdd8140c 100644 --- a/tuner/tuner/common.py +++ b/tuner/tuner/common.py @@ -163,7 +163,7 @@ def get_lowering_config( return iree_gpu.LoweringConfigAttr.get(lowering_config_attrs) -# Generate a config dictionary in translation info +# Generate a config dictionary used in translation_info attribute. def get_translation_info_config( pipeline_options: iree_gpu.PipelineOptionsAttr, waves_per_eu: int ) -> ir.DictAttr: @@ -177,16 +177,7 @@ def get_translation_info_config( llvm_func_attrs = {"amdgpu-waves-per-eu" = "3"} } > - Example Usage: - pipeline_options = iree_gpu.PipelineOptionsAttr.get(...) - waves_per_eu = 3 - - config_dict = get_translation_info_config( - pipeline_options=pipeline_options, - waves_per_eu=waves_per_eu - ) - - this 'config_dict' is subsequently used afterward to generate the 'translation_info' in the above example IR.""" + """ waves_per_eu_str = str(waves_per_eu) # Create the waves_per_eu dictionary attribute.