| #!/usr/bin/env vpython3 |
| # Copyright 2022 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| # Credits to the The Fuchsia Authors for creating this file. |
| """This script is used to fetch the tensorflow 2.7.0 pip package (via vpython) |
| for use by MLGO during LLVM compile. The vpython spec is hand-created, with the |
| help of pip. The transitive dependencies can be retrieved by using pipdeptree, |
| a pip package, by running `pipdeptree -p tensorflow`. |
| """ |
| # [VPYTHON:BEGIN] |
| # python_version: "3.8" |
| # wheel: < |
| # name: "infra/python/wheels/absl-py-py3" |
| # version: "version:2.1.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/astunparse-py2_py3" |
| # version: "version:1.6.3" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/cachetools-py3" |
| # version: "version:4.2.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/certifi-py2_py3" |
| # version: "version:2021.5.30" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/charset_normalizer-py3" |
| # version: "version:2.0.4" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/flatbuffers-py3" |
| # version: "version:24.3.25" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/gast-py3" |
| # version: "version:0.4.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/google-auth-oauthlib-py3" |
| # version: "version:1.0.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/google-auth-py3" |
| # version: "version:2.16.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/google-pasta-py3" |
| # version: "version:0.2.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/grpcio/${vpython_platform}" |
| # version: "version:1.57.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/h5py/${vpython_platform}" |
| # version: "version:3.11.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/idna-py3" |
| # version: "version:3.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/importlib-metadata-py3" |
| # version: "version:8.0.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/jax-py3" |
| # version: "version:0.4.13" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/keras-py3" |
| # version: "version:2.12.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/libclang/${vpython_platform}" |
| # version: "version:18.1.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/markdown-py3" |
| # version: "version:3.3.4" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/markupsafe/${vpython_platform}" |
| # version: "version:2.1.5" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/ml_dtypes/${vpython_platform}" |
| # version: "version:0.2.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/numpy/${vpython_platform}" |
| # version: "version:1.22.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/oauthlib-py2_py3" |
| # version: "version:3.2.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/opt-einsum-py3" |
| # version: "version:3.3.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/packaging-py3" |
| # version: "version:24.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/protobuf-py3" |
| # version: "version:4.25.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/pyasn1-py2_py3" |
| # version: "version:0.4.8" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/pyasn1_modules-py2_py3" |
| # version: "version:0.2.8" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/requests-py3" |
| # version: "version:2.31.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/requests-oauthlib-py2_py3" |
| # version: "version:2.0.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/rsa-py3" |
| # version: "version:4.7.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/scipy/${vpython_platform}" |
| # version: "version:1.10.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/setuptools-py3" |
| # version: "version:70.3.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/six-py2_py3" |
| # version: "version:1.16.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorboard-py3" |
| # version: "version:2.12.3" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorboard-data-server-py3" |
| # version: "version:0.7.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorflow/${vpython_platform}" |
| # version: "version:2.12.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorflow-estimator-py3" |
| # version: "version:2.12.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorflow-io-gcs-filesystem/${vpython_platform}" |
| # version: "version:0.34.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/termcolor-py2_py3" |
| # version: "version:2.4.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/typing-extensions-py3" |
| # version: "version:4.0.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/urllib3-py2_py3" |
| # version: "version:1.26.6" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/werkzeug-py3" |
| # version: "version:3.0.3" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/wheel-py2_py3" |
| # version: "version:0.37.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/wrapt/${vpython_platform}" |
| # version: "version:1.14.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/zipp-py3" |
| # version: "version:3.7.0" |
| # > |
| |
| # [VPYTHON:END] |
| import importlib |
| import os |
| |
| spec = importlib.util.find_spec("tensorflow") |
| print(os.path.dirname(spec.origin)) |
