Function reportAllocatedMemory

query/device_allocator.go:133–155

(deviceCount int, da deviceAllocator)

}

func reportAllocatedMemory(deviceCount int, da deviceAllocator) {
	// getAllocatedMemory may panic, therefore we should recover here
	defer func() {
		if r := recover(); r != nil {
			var err error
			switch x := r.(type) {
			case string:
				err = utils.StackError(nil, x)
			case error:
				err = utils.StackError(x, "Panic happens when reporting allocated memory")
			default:
				err = utils.StackError(nil, "Panic happens when reporting allocated memory %v", x)
			}
			utils.GetLogger().With("err", err).Error("Failed to report allocated memory")
		}
	}()

	for device := 0; device < deviceCount; device++ {
		utils.GetRootReporter().GetChildGauge(map[string]string{
			"device": strconv.Itoa(device),
		}, utils.AllocatedDeviceMemory).Update(float64(da.getAllocatedMemory(device)))
	}
}

// newDeviceAllocator returns a new device allocator instances.
func newDeviceAllocator() deviceAllocator {