Skip to content

Commit fb7c758

Browse files
committed
Fixes for building and publishing onnx for gpu.
1 parent daa6b2e commit fb7c758

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

opt/build/onnxruntime/Makefile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,10 +53,10 @@ IID_$(1)=$(1)_$(VERSION).iid
5353
CID_$(1)=$(1)_$(VERSION).cid
5454

5555
build_x64-gpu:
56-
@docker build --iidfile $$(IID_$(1)) -t redisfab/$(STEM)-x64:$(VERSION) -f Dockerfile.x64-gpu \
56+
@docker build --iidfile $$(IID_$(1)) -t redisfab/$(STEM)-$(1):$(VERSION) -f Dockerfile.x64-gpu \
5757
--build-arg OS=$(DOCKER_OS) $(ROOT)
5858
@docker create --cidfile $$(CID_$(1)) `cat $$(IID_$(1))`
59-
@docker cp `cat $$(CID_$(1))`:/build/$(STEM)-x64-$(VERSION).tgz .
59+
@docker cp `cat $$(CID_$(1))`:/build/$(STEM)-$(1)-$(VERSION).tgz .
6060

6161
.PHONY: build_x64
6262
endef

opt/build/onnxruntime/pack.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
set -e
44
VER="$1"
5-
PLATFORM=$($READIES/bin/platform -t)
5+
PLATFORM="$2"
66

77
mkdir -p pack/include pack/lib
88
cp onnxruntime/build/Linux/MinSizeRel/libonnxruntime.so.${VER} pack/lib/
@@ -19,5 +19,5 @@ cp onnxruntime/include/onnxruntime/core/providers/cuda/cuda_provider_factory.h p
1919
cd pack/lib/
2020
ln -s libonnxruntime.so.${VER} libonnxruntime.so
2121
cd ../..
22-
mv pack onnxruntime-${PLATFORM}-${VER}
23-
tar czf onnxruntime-${PLATFORM}-${VER}.tgz onnxruntime-${PLATFORM}-${VER}/
22+
mv pack onnxruntime-linux-${PLATFORM}-${VER}
23+
tar czf onnxruntime-linux-${PLATFORM}-${VER}.tgz onnxruntime-linux-${PLATFORM}-${VER}/

0 commit comments

Comments
 (0)