Commit d3bd1093 authored by Driss KHALIL's avatar Driss KHALIL

[Device.type] Fix add gpu constant error

parent 07dd798b
Pipeline #52727 passed with stage
in 21 minutes and 12 seconds
......@@ -138,7 +138,7 @@ def run(
shutil.move(static_logfile_name, backup)
with open(static_logfile_name, "w", newline="") as f:
logdata = cpu_constants()
if device == "cuda":
if device.type == "cuda":
logdata += gpu_constants()
logdata += (("model_size", n),)
logwriter = csv.DictWriter(f, fieldnames=[k[0] for k in logdata])
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment