Skip to content

Commit c7c3171

Browse files
committed
Re-use per-worker vision tensors as means for optimized vision benchmark
1 parent 19cd8a5 commit c7c3171

File tree

1 file changed

+5
-5
lines changed
  • cmd/aibench_run_inference_redisai_vision

1 file changed

+5
-5
lines changed

cmd/aibench_run_inference_redisai_vision/main.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@ var (
4040
)
4141

4242
// Vars only for git sha and diff handling
43-
var GitSHA1 string = ""
44-
var GitDirty string = "0"
43+
var GitSHA1 = ""
44+
var GitDirty = "0"
4545

4646
func AibenchGitSHA1() string {
4747
return GitSHA1
@@ -51,7 +51,7 @@ func AibenchGitDirty() (dirty bool) {
5151
dirty = false
5252
dirtyLines, err := strconv.Atoi(GitDirty)
5353
if err == nil {
54-
dirty = (dirtyLines != 0)
54+
dirty = dirtyLines != 0
5555
}
5656
return
5757
}
@@ -279,8 +279,8 @@ func (p *Processor) ProcessInferenceQuery(q []byte, isWarm bool, workerNum int,
279279
if isWarm && p.opts.showExplain {
280280
return nil, nil
281281
}
282-
tensorName := fmt.Sprintf("imageTensor:{w%d-i%d}", workerNum, queryNumber)
283-
outputTensorName := fmt.Sprintf("classificationTensor:{w%d-i%d}", workerNum, queryNumber)
282+
tensorName := fmt.Sprintf("imageTensor:{w%d}", workerNum)
283+
outputTensorName := fmt.Sprintf("classificationTensor:{w%d}", workerNum)
284284
tensorValues := q
285285
pos := rand.Int31n(int32(len(p.pclient)))
286286
var err error

0 commit comments

Comments
 (0)