Skip to content

Path Condition to ONNX interface #95

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
gsvgit merged 1 commit into from
Apr 8, 2025
Merged

Path Condition to ONNX interface #95

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
73 changes: 68 additions & 5 deletions VSharp.Explorer/AISearcher.fs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -213,7 +213,8 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai
aiAgentTrainingModelOptions: Option<AIAgentTrainingModelOptions>
) =
let numOfVertexAttributes = 7
let numOfStateAttributes = 7
let numOfStateAttributes = 6
let numOfPathConditionVertexAttributes = 49
let numOfHistoryEdgeAttributes = 2


Expand Down Expand Up @@ -248,10 +249,33 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai
let gameState = currentGameState.Value
let stateIds = Dictionary<uint<stateId>, int>()
let verticesIds = Dictionary<uint<basicBlockGlobalId>, int>()
let pathConditionVerticesIds = Dictionary<uint<pathConditionVertexId>, int>()

let networkInput =
let res = Dictionary<_, _>()

let pathConditionVertices, numOfPcToPcEdges =
let mutable numOfPcToPcEdges = 0

let shape =
[| int64 gameState.PathConditionVertices.Length
numOfPathConditionVertexAttributes |]

let attributes =
Array.zeroCreate (
gameState.PathConditionVertices.Length * numOfPathConditionVertexAttributes
)

for i in 0 .. gameState.PathConditionVertices.Length - 1 do
let v = gameState.PathConditionVertices.[i]
numOfPcToPcEdges <- numOfPcToPcEdges + v.Children.Length * 2
pathConditionVerticesIds.Add(v.Id, i)
let j = i * numOfPathConditionVertexAttributes
attributes.[j + int v.Type] <- float32 1u

OrtValue.CreateTensorValueFromMemory(attributes, shape), numOfPcToPcEdges


let gameVertices =
let shape = [| int64 gameState.GraphVertices.Length; numOfVertexAttributes |]

Expand Down Expand Up @@ -285,8 +309,6 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai
stateIds.Add(v.Id, i)
let j = i * numOfStateAttributes
attributes.[j] <- float32 v.Position
// TODO: Support path condition
// attributes.[j + 1] <- float32 v.PathConditionSize
attributes.[j + 2] <- float32 v.VisitedAgainVertices
attributes.[j + 3] <- float32 v.VisitedNotCoveredVerticesInZone
attributes.[j + 4] <- float32 v.VisitedNotCoveredVerticesOutOfZone
Expand All @@ -295,6 +317,31 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai

OrtValue.CreateTensorValueFromMemory(attributes, shape), numOfParentOfEdges, numOfHistoryEdges

let pcToPcEdgeIndex =
let shapeOfIndex = [| 2L; numOfPcToPcEdges |]
let index = Array.zeroCreate (2 * numOfPcToPcEdges)
let mutable firstFreePositionOfIndex = 0

for v in gameState.PathConditionVertices do
for child in v.Children do
// from vertex to child
index.[firstFreePositionOfIndex] <- pathConditionVerticesIds.[v.Id]

index.[firstFreePositionOfIndex + 2 * numOfPcToPcEdges] <-
pathConditionVerticesIds.[child]

firstFreePositionOfIndex <- firstFreePositionOfIndex + 1
// from child to vertex
index.[firstFreePositionOfIndex] <- pathConditionVerticesIds.[child]

index.[firstFreePositionOfIndex + 2 * numOfPcToPcEdges] <-
pathConditionVerticesIds.[v.Id]

firstFreePositionOfIndex <- firstFreePositionOfIndex + 1

OrtValue.CreateTensorValueFromMemory(index, shapeOfIndex)


let vertexToVertexEdgesIndex, vertexToVertexEdgesAttributes =
let shapeOfIndex = [| 2L; gameState.Map.Length |]
let shapeOfAttributes = [| int64 gameState.Map.Length |]
Expand All @@ -310,11 +357,13 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai
OrtValue.CreateTensorValueFromMemory(index, shapeOfIndex),
OrtValue.CreateTensorValueFromMemory(attributes, shapeOfAttributes)

let historyEdgesIndex_vertexToState, historyEdgesAttributes, parentOfEdges =
let historyEdgesIndex_vertexToState, historyEdgesAttributes, parentOfEdges, edgeIndex_pcToState =
let shapeOfParentOf = [| 2L; numOfParentOfEdges |]
let parentOf = Array.zeroCreate (2 * numOfParentOfEdges)
let shapeOfHistory = [| 2L; numOfHistoryEdges |]
let historyIndex_vertexToState = Array.zeroCreate (2 * numOfHistoryEdges)
let shapeOfPcToState = [| 2L; gameState.States.Length |]
let index_pcToState = Array.zeroCreate (2 * gameState.States.Length)

let shapeOfHistoryAttributes =
[| int64 numOfHistoryEdges; int64 numOfHistoryEdgeAttributes |]
Expand All @@ -323,6 +372,7 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai
let mutable firstFreePositionInParentsOf = 0
let mutable firstFreePositionInHistoryIndex = 0
let mutable firstFreePositionInHistoryAttributes = 0
let mutable firstFreePositionInPcToState = 0

gameState.States
|> Array.iter (fun state ->
Expand All @@ -334,6 +384,14 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai

firstFreePositionInParentsOf <- firstFreePositionInParentsOf + state.Children.Length

index_pcToState.[firstFreePositionInPcToState] <-
int64 pathConditionVerticesIds[state.PathCondition.Id]

index_pcToState.[firstFreePositionInPcToState + gameState.States.Length] <-
int64 stateIds[state.Id]

firstFreePositionInPcToState <- firstFreePositionInPcToState + 1

state.History
|> Array.iteri (fun i historyElem ->
let j = firstFreePositionInHistoryIndex + i
Expand All @@ -352,7 +410,8 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai

OrtValue.CreateTensorValueFromMemory(historyIndex_vertexToState, shapeOfHistory),
OrtValue.CreateTensorValueFromMemory(historyAttributes, shapeOfHistoryAttributes),
OrtValue.CreateTensorValueFromMemory(parentOf, shapeOfParentOf)
OrtValue.CreateTensorValueFromMemory(parentOf, shapeOfParentOf),
OrtValue.CreateTensorValueFromMemory(index_pcToState, shapeOfPcToState)

let statePosition_stateToVertex, statePosition_vertexToState =
let data_stateToVertex = Array.zeroCreate (2 * gameState.States.Length)
Expand Down Expand Up @@ -380,6 +439,7 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai

res.Add("game_vertex", gameVertices)
res.Add("state_vertex", states)
res.Add("path_condition_vertex", pathConditionVertices)

res.Add("gamevertex_to_gamevertex_index", vertexToVertexEdgesIndex)
res.Add("gamevertex_to_gamevertex_type", vertexToVertexEdgesAttributes)
Expand All @@ -390,6 +450,9 @@ type internal AISearcher(oracle: Oracle, aiAgentTrainingMode: Option<AIAgentTrai
res.Add("gamevertex_in_statevertex", statePosition_vertexToState)
res.Add("statevertex_parentof_statevertex", parentOfEdges)

res.Add("pathconditionvertex_to_pathconditionvertex", pcToPcEdgeIndex)
res.Add("pathconditionvertex_to_statevertex", edgeIndex_pcToState)

res

let output = session.Run(runOptions, networkInput, session.OutputNames)
Expand Down
Loading