hw2vec

简介：
HW2Vec是一个基于Python的自动化设计库，旨在简化硬件设计的流程。该仓库整合了两个主要项目：首先，利用图学习技术（RTP）在RTL和门级网表中识别潜在的硬件木马；其次，通过图学习（GNN4IP）进行硬件设计中的IP盗版检测，具体而言是克隆检测。我们巧妙地运用了PyHDI提供的图形提取管道的一部分功能。 该库具备从Verilog代码中提取数据流图和控制流图的能力。 典型的管道架构流程如下：项目结构如下所示：临时目录结构如图所示。 核心代码位于hw2vec文件夹中：hw2vec/hw2graph.py包含用于从硬件设计生成图形对象的代码；hw2vec/graph2vec则包含了图训练模型和教练相关的代码。 为了方便用户理解，提供了示例用例，这些示例存储在examples文件夹中：examples/gnn4ip_netlist.py展示了在网表数据集上运行的gnn4ip脚本；examples/gnn4ip_RTL.py则提供了在RTL数据集上运行的gnn4i脚本。