Remove .oss from find_cuda_config in compression script.
See https://github.com/tensorflow/tensorflow/pull/40759
PiperOrigin-RevId: 318452377
Change-Id: I04f3ad1c8cf9cac5446d0a1196ebbf66660bf312
diff --git a/third_party/gpus/compress_find_cuda_config.py b/third_party/gpus/compress_find_cuda_config.py
index 98be39d..606bbf2 100644
--- a/third_party/gpus/compress_find_cuda_config.py
+++ b/third_party/gpus/compress_find_cuda_config.py
@@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================
-"""Compresses the contents of find_cuda_config.py.oss.
+"""Compresses the contents of 'find_cuda.py'.
The compressed file is what is actually being used. It works around remote
config not being able to upload files yet.
@@ -22,13 +22,13 @@
def main():
- with open('find_cuda_config.py.oss', 'rb') as f:
+ with open('find_cuda.py', 'rb') as f:
data = f.read()
compressed = zlib.compress(data)
b64encoded = base64.b64encode(compressed)
- with open('find_cuda_config.py.gz.base64.oss', 'wb') as f:
+ with open('find_cuda.py.gz.base64', 'wb') as f:
f.write(b64encoded)
