fix: Incorrect cache ID directory (#251)

tensorchord · Jun 9, 2022 · 15d9c51 · 15d9c51
1 parent 8f8c81e
commit 15d9c51
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/lang/ir/cache.go b/pkg/lang/ir/cache.go
@@ -24,9 +24,9 @@ func (g Graph) CacheID(filename string) string {
  gpu := g.CUDA != nil || g.CUDNN != nil
  var cacheID string
  if gpu {
- cacheID = fmt.Sprintf("/%s-gpu-%s", g.CachePrefix, filename)
+ cacheID = fmt.Sprintf("%s/%s-gpu", filename, g.CachePrefix)
  } else {
- cacheID = fmt.Sprintf("/%s-cpu-%s", g.CachePrefix, filename)
+ cacheID = fmt.Sprintf("%s/%s-cpu", filename, g.CachePrefix)
  }
  logrus.Debugf("apt/pypi calculated cacheID: %s", cacheID)
  return cacheID