mirror of
https://github.com/hanzo-ml/kserve.git
synced 2026-10-30 20:53:49 +00:00
SivananthamandGitHub
4b84d130dd
* replace github.com/golang/protobuf Signed-off-by: Sivanantham Chinnaiyan <[email protected]> * regenerate tf2openapi Signed-off-by: Sivanantham Chinnaiyan <[email protected]> * remove use of github.com/gogo/protobuf pkg Signed-off-by: Sivanantham Chinnaiyan <[email protected]> --------- Signed-off-by: Sivanantham Chinnaiyan <[email protected]>
45 lines
1.8 KiB
Makefile
45 lines
1.8 KiB
Makefile
VENDOR_DIR := $(GOPATH)/src/github.com/kserve/kserve/vendor
|
|
TF_PROTO_SRC := $(VENDOR_DIR)/github.com/tensorflow/tensorflow
|
|
TF_PROTO_OUT := generated
|
|
|
|
# Run tests
|
|
test: generate
|
|
go test ./types/... ./generator/...
|
|
go build -o bin/tf2openapi ./cmd/main.go
|
|
go test ./cmd/...
|
|
|
|
# Generate code
|
|
generate:
|
|
ifndef GOPATH
|
|
$(error GOPATH not defined, please define GOPATH. Run "go help gopath" to learn more about GOPATH)
|
|
endif
|
|
|
|
ifeq (,$(findstring $(GOPATH)/bin, $(PATH)))
|
|
$(error GOPATH bin not in path, please add it to PATH)
|
|
endif
|
|
|
|
# compile TensorFlow protos
|
|
TEMP_DIR=$$(mktemp -d /tmp/tf2openapi.XXX); \
|
|
protoc -I $(TF_PROTO_SRC) --go_out=$$TEMP_DIR $(TF_PROTO_SRC)/tensorflow/core/lib/core/error_codes.proto; \
|
|
protoc -I $(TF_PROTO_SRC) --go_out=$$TEMP_DIR $(TF_PROTO_SRC)/tensorflow/core/framework/*.proto; \
|
|
protoc -I $(TF_PROTO_SRC) --go_out=$$TEMP_DIR $(TF_PROTO_SRC)/tensorflow/core/protobuf/saver.proto; \
|
|
protoc -I $(TF_PROTO_SRC) --go_out=$$TEMP_DIR $(TF_PROTO_SRC)/tensorflow/core/protobuf/meta_graph.proto; \
|
|
protoc -I $(TF_PROTO_SRC) --go_out=$$TEMP_DIR $(TF_PROTO_SRC)/tensorflow/core/protobuf/saved_model.proto; \
|
|
if [ -d "./$(TF_PROTO_OUT)" ]; then \
|
|
rm -rf $(TF_PROTO_OUT); \
|
|
fi; \
|
|
mkdir $(TF_PROTO_OUT); \
|
|
for dir in lib framework protobuf; do \
|
|
echo "$$dir"; \
|
|
if [ ! -d "./$(TF_PROTO_OUT)/$$dir" ]; then \
|
|
mkdir $(TF_PROTO_OUT)/$$dir; \
|
|
fi; \
|
|
mv $$TEMP_DIR/github.com/tensorflow/tensorflow/tensorflow/go/core/$$dir/* ./$(TF_PROTO_OUT)/$$dir; \
|
|
done; \
|
|
rm -rf $$TEMP_DIR
|
|
|
|
# update meta_graph.pb.go to import generated package
|
|
perl -pi -e 's+github.com/tensorflow/tensorflow/tensorflow/go/core/framework+github.com/kserve/kserve/tools/tf2openapi/$(TF_PROTO_OUT)/framework+g' \
|
|
./generated/protobuf/meta_graph.pb.go
|
|
|