13 lines
376 B
Smarty
13 lines
376 B
Smarty
__version__ = '{{VERSION}}'
|
|
debug = False
|
|
cuda = '{{CUDA_VERSION}}'
|
|
# TODO: use workspace status to stamp the correct version
|
|
git_version = ""
|
|
hip = None
|
|
|
|
# This is a gross monkey-patch hack that depends on the order of imports
|
|
# in torch/__init__.py
|
|
# TODO: find a more elegant solution to set `USE_GLOBAL_DEPS` for the bazel build
|
|
import torch
|
|
torch.USE_GLOBAL_DEPS = False
|