Skip to content

Commit 3112c79

Browse files
Fix CPU stats
Container stats report cumulative CPU secs from the container was started. Signed-off-by: andreasjansson <[email protected]>
1 parent f4b54b4 commit 3112c79

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

pkg/serving/local.go

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import (
66
"encoding/json"
77
"fmt"
88
"io"
9+
"math/rand"
910
"mime/multipart"
1011
"net/http"
1112
"os"
@@ -66,7 +67,7 @@ func (p *LocalDockerPlatform) Deploy(imageTag string, logWriter logger.Logger) (
6667
}
6768
*/
6869

69-
hostPort, err := shell.NextFreePort(5000)
70+
hostPort, err := shell.NextFreePort(5000 + rand.Intn(1000))
7071
if err != nil {
7172
return nil, err
7273
}
@@ -193,6 +194,12 @@ func (d *LocalDockerDeployment) RunInference(input *Example, logWriter logger.Lo
193194
if err := mwriter.Close(); err != nil {
194195
return nil, fmt.Errorf("Failed to close form mime writer: %w", err)
195196
}
197+
198+
_, usedCPUSecsStart, err := d.getResourceUsage()
199+
if err != nil {
200+
return nil, err
201+
}
202+
196203
url := fmt.Sprintf("http://localhost:%d/infer", d.port)
197204
req, err := http.NewRequest(http.MethodPost, url, bodyBuffer)
198205
if err != nil {
@@ -209,10 +216,11 @@ func (d *LocalDockerDeployment) RunInference(input *Example, logWriter logger.Lo
209216
}
210217
defer resp.Body.Close()
211218

212-
usedMemoryBytes, usedCPUSecs, err := d.getResourceUsage()
219+
usedMemoryBytes, usedCPUSecsEnd, err := d.getResourceUsage()
213220
if err != nil {
214221
return nil, err
215222
}
223+
usedCPUSecs := usedCPUSecsEnd - usedCPUSecsStart
216224

217225
if resp.StatusCode == http.StatusBadRequest {
218226
body := struct {

0 commit comments

Comments
 (0)