Add sh_binary rule to copy tensorflow wheel to wheel artifact directory

ROCm · Nov 5, 2024 · 33a418c · 33a418c
1 parent 554f81b
commit 33a418c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tensorflow/tools/pip_package/BUILD b/tensorflow/tools/pip_package/BUILD
@@ -315,6 +315,13 @@ genrule(
     cmd = "echo '' > $@",
 )
 
+
+sh_binary(
+    name = "copy_wheel",
+    srcs = ['copy_wheel.sh'],
+    data = [":wheel"],
+)
+
 py_test(
     name = "prebuilt_wheel_import_api_packages_test",
     srcs = if_wheel_dependency(

diff --git a/tensorflow/tools/pip_package/copy_wheel.sh b/tensorflow/tools/pip_package/copy_wheel.sh
@@ -0,0 +1,3 @@
+#!/bin/bash
+
+cp  tensorflow/tools/pip_package/wheel_house/*.whl $1
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		#!/bin/bash

		cp tensorflow/tools/pip_package/wheel_house/*.whl $1