68 lines
2.6 KiB
Python
68 lines
2.6 KiB
Python
|
import json
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
# ================= 配置 =================
|
|||
|
|
# 数据集路径 (请确保指向你正在使用的那个文件)
|
|||
|
|
DATASET_PATH = "data/HDLBits/HDLBits_data.jsonl"
|
|||
|
|
# 提取出的 RTL 存放目录 (建议直接放到 saves 方便一起找)
|
|||
|
|
OUTPUT_BASE_DIR = "saves"
|
|||
|
|
|
|||
|
|
def extract_dut(task_id_target):
|
|||
|
|
found = False
|
|||
|
|
|
|||
|
|
if not os.path.exists(DATASET_PATH):
|
|||
|
|
print(f"❌ 错误：找不到数据集文件 {DATASET_PATH}")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
print(f"🔍 正在从 {DATASET_PATH} 中查找任务: {task_id_target} ...")
|
|||
|
|
|
|||
|
|
with open(DATASET_PATH, 'r', encoding='utf-8') as f:
|
|||
|
|
for line in f:
|
|||
|
|
try:
|
|||
|
|
data = json.loads(line)
|
|||
|
|
current_task_id = data.get('task_id')
|
|||
|
|
|
|||
|
|
if current_task_id == task_id_target:
|
|||
|
|
# 找到目标任务
|
|||
|
|
found = True
|
|||
|
|
|
|||
|
|
# 尝试获取标准代码
|
|||
|
|
# HDLBits 数据集通常用 'module_code' 或 'solution' 存标准答案
|
|||
|
|
dut_code = data.get('module_code') or data.get('solution')
|
|||
|
|
|
|||
|
|
if not dut_code:
|
|||
|
|
print(f"❌ 错误：在数据集里找到了任务，但 'module_code' 字段为空！")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 确定保存路径
|
|||
|
|
save_dir = os.path.join(OUTPUT_BASE_DIR, task_id_target)
|
|||
|
|
if not os.path.exists(save_dir):
|
|||
|
|
os.makedirs(save_dir)
|
|||
|
|
print(f"⚠️ 注意：创建了新目录 {save_dir}")
|
|||
|
|
|
|||
|
|
# 写入文件
|
|||
|
|
# 文件名通常命名为 prob_<task_id>.v
|
|||
|
|
filename = f"prob_{task_id_target}.v"
|
|||
|
|
file_path = os.path.join(save_dir, filename)
|
|||
|
|
|
|||
|
|
with open(file_path, 'w', encoding='utf-8') as vf:
|
|||
|
|
vf.write(dut_code)
|
|||
|
|
|
|||
|
|
print(f"✅ 成功！DUT 文件已提取到：")
|
|||
|
|
print(f" -> {file_path}")
|
|||
|
|
print("\n现在你可以运行 Verilator 了：")
|
|||
|
|
print(f"verilator --binary --timing --coverage -j 0 {filename} ...")
|
|||
|
|
break
|
|||
|
|
except Exception as e:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
if not found:
|
|||
|
|
print(f"❌ 未找到任务 ID: {task_id_target}")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
if len(sys.argv) < 2:
|
|||
|
|
print("用法: python extract_dut.py <任务ID>")
|
|||
|
|
print("示例: python extract_dut.py fsm_serial")
|
|||
|
|
else:
|
|||
|
|
extract_dut(sys.argv[1])
|