register.go

package caffe

import (
	"os"

	assetfs "github.com/elazarl/go-bindata-assetfs"
	"github.com/c3sr/dlframework"
	"github.com/c3sr/dlframework/framework"
)

// FrameworkManifest ...
var FrameworkManifest = dlframework.FrameworkManifest{
	Name:    "Caffe",
	Version: "1.0",
	Container: map[string]*dlframework.ContainerHardware{
		"amd64": {
			Cpu: "raiproject/carml-caffe:amd64-cpu",
			Gpu: "raiproject/carml-caffe:amd64-gpu",
		},
		"ppc64le": {
			Cpu: "raiproject/carml-caffe:ppc64le-gpu",
			Gpu: "raiproject/carml-caffe:ppc64le-gpu",
		},
	},
}

func assetFS() *assetfs.AssetFS {
	assetInfo := func(path string) (os.FileInfo, error) {
		return os.Stat(path)
	}
	for k := range _bintree.Children {
		return &assetfs.AssetFS{Asset: Asset, AssetDir: AssetDir, AssetInfo: assetInfo, Prefix: k}
	}
	panic("unreachable")
}

func Register() {
	framework.Register(FrameworkManifest, assetFS())
}