()
| 213 | onnx_path = os.path.join(clip_local_dir, onnx_file) |
| 214 | |
| 215 | def get_compiled_file_name(): |
| 216 | name = f"model_b{bs}" |
| 217 | if fp16: |
| 218 | name += "_fp16" |
| 219 | elif bf16: |
| 220 | name += "_bf16" |
| 221 | |
| 222 | if exhaustive_tune: name += "_exh" |
| 223 | return name + ".mxr" |
| 224 | |
| 225 | clip_compiled_dir = get_local_path(compiled_dir, model_dir) |
| 226 | mxr_file = get_compiled_file_name() |
no outgoing calls
no test coverage detected