rename variables

sarchlab · Nov 12, 2024 · 02bf564 · 02bf564
1 parent b30554f
commit 02bf564
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 11 deletions.
diff --git a/driver/builder.go b/driver/builder.go
@@ -17,7 +17,7 @@ type Builder struct {
 	useMagicMemoryCopy  bool
 	middlewareD2HCycles int
 	middlewareH2DCycles int
-	memorySize          uint64
+	cpuMemorySize       uint64
 }
 
 // MakeBuilder creates a driver builder with some default configuration
@@ -76,7 +76,7 @@ func (b Builder) WithH2DCycles(h2dCycles int) Builder {
 }
 
 // WithMemorySize sets the memory size of the CPU.
-func (b Builder) WithMemorySize(memorySize uint64) Builder {
+func (b Builder) WithCPUMemorySize(memorySize uint64) Builder {
 	b.memorySize = memorySize
 	return b
 }

diff --git a/driver/driver.go b/driver/driver.go
@@ -1,7 +1,6 @@
 package driver
 
 import (
-	"fmt"
 	"log"
 	"reflect"
 	"runtime/debug"
@@ -462,9 +461,6 @@ func (d *Driver) distributeWGToGPUs(
 		panic("not all wg allocated")
 	}
 
-	// fmt.Sprintln("total WG: %d WG Per CU %d\n", totalWGCount, wgPerCU)
-	fmt.Printf("total WG: %d WG Per CU %d\n", totalWGCount, wgPerCU)
-
 	return wgDist
 }
 

diff --git a/driver/internal/memoryallocator.go b/driver/internal/memoryallocator.go
@@ -65,9 +65,7 @@ func (a *memoryAllocatorImpl) RegisterDevice(device *Device) {
 
 	state := device.MemState
 	state.setInitialAddress(a.totalStorageByteSize)
-
 	a.totalStorageByteSize += state.getStorageSize()
-
 	a.devices[device.ID] = device
 }
 

diff --git a/timing/cp/builder.go b/timing/cp/builder.go
@@ -86,9 +86,9 @@ func (b Builder) Build(name string) *CommandProcessor {
 		tracing.CollectTrace(cp, b.visTracer)
 	}
 
-	// if b.perfAnalyzer != nil {
-	// 	b.perfAnalyzer.RegisterComponent(cp)
-	// }
+	if b.perfAnalyzer != nil {
+		b.perfAnalyzer.RegisterComponent(cp)
+	}
 
 	return cp
 }