add a tiny benchmark (#369)

* tiny benchmark * Update fakemodel.go
lsds · Feb 23, 2024 · b1e1768 · b1e1768
1 parent 80dfa46
commit b1e1768
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tests/go/fakemodel/fakemodel.go b/tests/go/fakemodel/fakemodel.go
@@ -14,6 +14,7 @@ var Models = map[string][]int{
 	"vgg16-imagenet":    vgg16Imagenet,
 	"slp-mnist":         slpMNIST,
 	"bert":              bert,
+	"tiny":              genFakeModel(8, 10000),
 }
 
 var Names = func(m map[string][]int) []string {

diff --git a/tests/go/fakemodel/tiny.go b/tests/go/fakemodel/tiny.go
@@ -0,0 +1,9 @@
+package fakemodel
+
+func genFakeModel(k, n int) []int {
+	a := make([]int, n)
+	for i := range a {
+		a[i] = k
+	}
+	return a
+}