Add type cache of linked programs.

Author: RZhang05

bbq/vm/linker.go

@@ -49,6 +49,8 @@ func LinkGlobals(
 	globals := make([]Variable, len(program.Globals))
 	indexedGlobals := activations.NewActivation[Variable](memoryGauge, nil)
 
+	var visitedLocations map[common.Location]struct{}
+
 	// NOTE: ensure both the context and the mapping are updated
 
 	for _, global := range program.Globals {
@@ -108,6 +110,7 @@ func LinkGlobals(
 				typedGlobal,
 				context,
 				linkedGlobalsCache,
+				visitedLocations,
 			)
 			globals[index] = importedGlobal
 
@@ -145,6 +148,7 @@ func linkImportedGlobal(
 	importedGlobal *bbq.ImportedGlobal,
 	context *Context,
 	linkedGlobalsCache map[common.Location]LinkedGlobals,
+	visitedLocations map[common.Location]struct{},
 ) Variable {
 	importLocation := importedGlobal.Location
 
@@ -171,7 +175,17 @@ func linkImportedGlobal(
 				linkedGlobalsCache,
 			)
 
-			// TODO: Potentially add the imported program's sema type cache to the context's sema type cache
+			if _, ok := visitedLocations[importLocation]; !ok {
+				// add the imported program's sema type cache to the context's sema type cache
+				// TODO: Maybe this is excessive, balance size and performance
+				for typeID, semaType := range importedProgram.SemaTypeCache {
+					if _, ok := context.semaTypeCache[typeID]; ok {
+						continue
+					}
+					context.semaTypeCache[typeID] = semaType
+				}
+				visitedLocations[importLocation] = struct{}{}
+			}
 		}
 
 		indexedGlobals = linkedGlobals.indexedGlobals