-
Notifications
You must be signed in to change notification settings - Fork 71
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(query): support query for difftest-dpic data (#557)
- Loading branch information
Showing
10 changed files
with
320 additions
and
8 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,113 @@ | ||
| /*************************************************************************************** | ||
| * Copyright (c) 2025 Beijing Institute of Open Source Chip (BOSC) | ||
| * Copyright (c) 2025 Institute of Computing Technology, Chinese Academy of Sciences | ||
| * | ||
| * DiffTest is licensed under Mulan PSL v2. | ||
| * You can use this software according to the terms and conditions of the Mulan PSL v2. | ||
| * You may obtain a copy of Mulan PSL v2 at: | ||
| * http://license.coscl.org.cn/MulanPSL2 | ||
| * | ||
| * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, | ||
| * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, | ||
| * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE. | ||
| * | ||
| * See the Mulan PSL v2 for more details. | ||
| ***************************************************************************************/ | ||
|
|
||
| package difftest.util | ||
|
|
||
| import difftest.DifftestBundle | ||
| import difftest.common.FileControl | ||
|
|
||
| import scala.collection.mutable.ListBuffer | ||
|
|
||
| object Query { | ||
| private val tables = ListBuffer.empty[QueryTable] | ||
| def register(gen: DifftestBundle, locPrefix: String) = { | ||
| tables += new QueryTable(gen, locPrefix) | ||
| } | ||
| def register(gens: Seq[DifftestBundle], locPrefix: String) = { | ||
| gens.foreach { gen => tables += new QueryTable(gen, locPrefix) } | ||
| } | ||
| def writeInvoke(gen: DifftestBundle): String = { | ||
| tables.find(_.gen == gen).get.writeInvoke | ||
| } | ||
| def collect() = { | ||
| val queryCpp = ListBuffer.empty[String] | ||
| queryCpp += | ||
| s""" | ||
| |#ifndef __DIFFTEST_QUERY_H__ | ||
| |#define __DIFFTEST_QUERY_H__ | ||
| | | ||
| |#include <cstdint> | ||
| |#include "diffstate.h" | ||
| |#include "query.h" | ||
| | | ||
| |#ifdef CONFIG_DIFFTEST_QUERY | ||
| | | ||
| |class QueryStats: public QueryStatsBase { | ||
| |public: | ||
| | ${tables.map { t => s"Query* ${t.instName};" }.mkString("\n ")} | ||
| | QueryStats(char *path): QueryStatsBase(path) { | ||
| | ${tables.map(_.initInvoke).mkString("\n ")} | ||
| | } | ||
| | ${tables.map(_.initDecl).mkString("")} | ||
| | ${tables.map(_.writeDecl).mkString("")} | ||
| |}; | ||
| |#endif // CONFIG_DIFFTEST_QUERY | ||
| |#endif // __DIFFTEST_QUERY_H__ | ||
| |""".stripMargin | ||
| FileControl.write(queryCpp, "difftest-query.h") | ||
| } | ||
| } | ||
|
|
||
| class QueryTable(val gen: DifftestBundle, locPrefix: String) { | ||
| val tableName: String = gen.desiredModuleName.replace("Difftest", "") | ||
| // Args: (key, value) | ||
| private val stepArgs: Seq[(String, String)] = Seq(("STEP", "query_step")) | ||
| private val locArgs: Seq[(String, String)] = { | ||
| val argList = ListBuffer.empty[(String, String)] | ||
| argList += (("COREID", "coreid")) | ||
| // resolve conflict with sql key | ||
| if (gen.isIndexed) argList += (("MY_INDEX", "index")) | ||
| if (gen.isFlatten) argList += (("ADDRESS", "address")) | ||
| argList.toSeq | ||
| } | ||
| private val dataArgs: Seq[(String, String)] = { | ||
| val dataPrefix = "packet->" | ||
| val argList = ListBuffer.empty[(String, String)] | ||
| for ((name, _, elem) <- gen.dataElements) { | ||
| val isRemoved = gen.isFlatten && Seq("valid", "address").contains(name) | ||
| if (!isRemoved) { | ||
| if (elem.length == 1) argList += ((name.toUpperCase, dataPrefix + name)) | ||
| else | ||
| Seq.tabulate(elem.length) { idx => | ||
| argList += (((name + s"_$idx").toUpperCase, dataPrefix + name + s"[$idx]")) | ||
| } | ||
| } | ||
| } | ||
| argList.toSeq | ||
| } | ||
| private val sqlArgs: Seq[(String, String)] = stepArgs ++ locArgs ++ dataArgs | ||
|
|
||
| val instName: String = "query_" + tableName | ||
| val initDecl = | ||
| s""" | ||
| | void ${tableName}_init() { | ||
| | const char* createSql = \" CREATE TABLE $tableName(\" \\ | ||
| | "ID INTEGER PRIMARY KEY AUTOINCREMENT," \\ | ||
| | ${sqlArgs.map("\"" + _._1 + " INT NOT NULL").mkString("", ",\" \\\n ", ");\";")} | ||
| | const char* insertSql = \"INSERT INTO $tableName (${sqlArgs.map(_._1).mkString(",")}) \" \\ | ||
| | \" VALUES (${Seq.fill(sqlArgs.length) { "?" }.mkString(",")});\"; | ||
| | $instName = new Query(mem_db, createSql, insertSql); | ||
| | } | ||
| |""".stripMargin | ||
| val initInvoke = s"${tableName}_init();" | ||
| val writeDecl = | ||
| s""" | ||
| | void ${tableName}_write(${locArgs.map("uint8_t " + _._2).mkString(", ")}, ${gen.desiredModuleName}* packet) { | ||
| | query_${tableName}->write(${sqlArgs.length}, ${sqlArgs.map(_._2).mkString(", ")}); | ||
| | } | ||
| |""".stripMargin | ||
| val writeInvoke = s"qStats->${tableName}_write(${locArgs.map(locPrefix + _._2).mkString(", ")}, packet);" | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,43 @@ | ||
| /*************************************************************************************** | ||
| * Copyright (c) 2025 Beijing Institute of Open Source Chip (BOSC) | ||
| * Copyright (c) 2025 Institute of Computing Technology, Chinese Academy of Sciences | ||
| * | ||
| * DiffTest is licensed under Mulan PSL v2. | ||
| * You can use this software according to the terms and conditions of the Mulan PSL v2. | ||
| * You may obtain a copy of Mulan PSL v2 at: | ||
| * http://license.coscl.org.cn/MulanPSL2 | ||
| * | ||
| * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, | ||
| * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, | ||
| * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE. | ||
| * | ||
| * See the Mulan PSL v2 for more details. | ||
| ***************************************************************************************/ | ||
|
|
||
| #ifdef CONFIG_DIFFTEST_QUERY | ||
| #include "query.h" | ||
| #include "difftest-query.h" | ||
|
|
||
| QueryStats *qStats; | ||
|
|
||
| void difftest_query_init() { | ||
| char query_path[128]; | ||
| snprintf(query_path, 128, "%s/build/%s", getenv("NOOP_HOME"), "difftest_query.db"); | ||
| // remove exist file | ||
| FILE *fp = fopen(query_path, "r"); | ||
| if (fp) { | ||
| fclose(fp); | ||
| remove(query_path); | ||
| } | ||
| qStats = new QueryStats(query_path); | ||
| } | ||
|
|
||
| void difftest_query_step() { | ||
| qStats->step(); | ||
| } | ||
|
|
||
| void difftest_query_finish() { | ||
| delete qStats; | ||
| } | ||
|
|
||
| #endif // CONFIG_DIFFTEST_QUERY |
Oops, something went wrong.