#!/bin/bash

# PyTorch Profiler 产生的 trace 文件如果包含非 UTF-8 的 stack 信息，
# 可以使用这个脚本强制修复。

INPUT_FILE="$1"
# 临时文件
TEMP_FILE="${INPUT_FILE}.temp.gz"

gzip -dc "$INPUT_FILE" | iconv -f utf-8 -t utf-8 -c | gzip > "$TEMP_FILE"

# 检查是否成功
if [ $? -eq 0 ]; then
    echo "Overwriting original file..."
    mv "$TEMP_FILE" "$INPUT_FILE"
    echo "Done."
else
    echo "Error, original file not overwritten."
    rm -f "$TEMP_FILE"
    exit 1
fi