mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixes for mypy

Browse Source
This commit is contained in:
Abhinav Goel 2024-10-02 13:46:10 -07:00 committed by GitHub
parent ec7a7791aa
commit ea7ad92a86
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/tools/pgo_nsys_converter.py
View File

@ -54,7 +54,7 @@ if __name__ == '__main__':
proc.wait()
thunk_re = re.compile("hlo_op=(.*)#")
cost_dictionary = dict()
cost_dictionary: dict[str, list] = dict()
with open(f"{args.pgle_output_path}", 'w', newline='') as protofile:
with open(f"{pgle_folder}{pgle_filename}.pbtxt_{report_name}.csv", newline='') as csvfile:
reader = csv.DictReader(csvfile)