From e8a28d8b67fed52c494e684561074dd1268cb49e Mon Sep 17 00:00:00 2001
From: Alex Vasile <48962821+Alex-Vasile@users.noreply.github.com>
Date: Wed, 24 Jun 2026 12:36:40 -0400
Subject: [PATCH] Fix dtypes in utils

---
 .../how-to/how-to-use-hipblaslt-tuning-utility.rst     |  6 +++---
 projects/hipblaslt/utilities/find_exact.py             | 10 ++++++----
 projects/hipblaslt/utilities/template.yaml             |  6 +++---
 3 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/projects/hipblaslt/docs/how-to/how-to-use-hipblaslt-tuning-utility.rst b/projects/hipblaslt/docs/how-to/how-to-use-hipblaslt-tuning-utility.rst
index f772e7b9e75a..3b9d1902579e 100644
--- a/projects/hipblaslt/docs/how-to/how-to-use-hipblaslt-tuning-utility.rst
+++ b/projects/hipblaslt/docs/how-to/how-to-use-hipblaslt-tuning-utility.rst
@@ -27,9 +27,9 @@ input for the ``find_exact.py`` script, as described below.
         DataTypeB: s
         DataTypeC: s
         DataTypeD: s
-        TransposeA: 0
-        TransposeB: 0
-        UseBias: False
+        TransposeA: false
+        TransposeB: false
+        UseBias: 0
     TestConfig:
         ColdIter: 20
         Iter: 100  # You can change this to a larger value for a more stable result, but the executing time also increases.
diff --git a/projects/hipblaslt/utilities/find_exact.py b/projects/hipblaslt/utilities/find_exact.py
index d87694522364..c6d633c49144 100644
--- a/projects/hipblaslt/utilities/find_exact.py
+++ b/projects/hipblaslt/utilities/find_exact.py
@@ -47,6 +47,8 @@
     from yaml import SafeLoader as yamlLoader
     assert 0 and "CSafeLoader not installed. Fallback to SafeLoader."
 
+from fix_yaml_types import fix_content
+
 #####################################################
 # Parameters
 #####################################################
@@ -59,13 +61,13 @@
 globalParameters["MatchTablePath"] = "/device-library/MatchTable.yaml"
 
 defaultBenchOptions = {"ProblemType": {
-    "TransposeA": 0,
-    "TransposeB": 0,
+    "TransposeA": False,
+    "TransposeB": False,
     "ComputeInputDataType": "s",
     "ComputeDataType": "s",
     "DataTypeC": "s",
     "DataTypeD": "s",
-    "UseBias": False
+    "UseBias": 0
 }, "TestConfig": {
     "ColdIter": 20,
     "Iter": 100,
@@ -115,7 +117,7 @@ def writeYAML(filename, data, **kwargs):
         kwargs["default_flow_style"] = None
 
     with open(filename, "w") as f:
-        yaml.dump(data, f, **kwargs)
+        f.write(fix_content(yaml.dump(data, **kwargs)))
 #########
 
 def dataType2Bench(dataType):
diff --git a/projects/hipblaslt/utilities/template.yaml b/projects/hipblaslt/utilities/template.yaml
index cd1876167b00..801080ea6136 100644
--- a/projects/hipblaslt/utilities/template.yaml
+++ b/projects/hipblaslt/utilities/template.yaml
@@ -7,9 +7,9 @@ Bench:
     DataTypeB: s
     DataTypeC: s
     DataTypeD: s
-    TransposeA: 0
-    TransposeB: 0
-    UseBias: False
+    TransposeA: false
+    TransposeB: false
+    UseBias: 0
   TestConfig:
     ColdIter: 20
     Iter: 100