predictor_disabled.go

// +build !ppc64le
// +build nogpu
// +build cgo

package tensorrt

import "C"
import (
	"context"

	"github.com/pkg/errors"
	"github.com/rai-project/dlframework/framework/options"
)

var (
	invalidSystemError = errors.New("invalid system. TensorRT is only available on linux systems")
)

type Predictor struct {
}

func New(ctx context.Context, opts0 ...options.Option) (*Predictor, error) {
	return nil, invalidSystemError
}

func (p *Predictor) Predict(ctx context.Context, input []float32) error {
	return invalidSystemError
}

func (p *Predictor) ReadPredictionOutput(ctx context.Context) ([]float32, error) {
	return nil, nil
}

func (p *Predictor) StartProfiling(name, metadata string) error {
	return invalidSystemError
}

func (p *Predictor) EndProfiling() error {
	return invalidSystemError
}

func (p *Predictor) DisableProfiling() error {
	return invalidSystemError
}

func (p *Predictor) ReadProfile() (string, error) {
	return "", invalidSystemError
}

func (p Predictor) Close() {

}