b7796f3c85
repository_ctx.execute() does not support uploading of files from the source tree. I initially tried constructing a command that simply embeds the file's contents. However that did not work on Windows because the file is larger than 8192 characters. So my best idea was to compress it locally and embed the compressed contents in the command and to uncompress it remotely. This works but comes with the drawback that we need to compress it first. This can't be done as part of the repository_rule either because within one repository_rule every execute() runs either locally or remotely. I thus decided to check in the compressed version in the source tree. It's very much a temporary measure as I'll add the ability to upload files to a future version of Bazel. PiperOrigin-RevId: 295787408 Change-Id: I1545dd86cdec7e4b20cba43d6a134ad6d1a08109
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
"""Compresses the contents of find_cuda_config.py.oss.
|
|
|
|
The compressed file is what is actually being used. It works around remote
|
|
config not being able to upload files yet.
|
|
"""
|
|
import base64
|
|
import zlib
|
|
|
|
|
|
def main():
|
|
with open('find_cuda_config.py.oss', '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:
|
|
f.write(b64encoded)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|