mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
24ca7e91e6
Turn more functions and variables into static if they are not used outside the cpp files. Unused functions are removed. Pull Request resolved: https://github.com/pytorch/pytorch/pull/150930 Approved by: https://github.com/Skylion007 Co-authored-by: Aaron Gokaslan <aaronGokaslan@gmail.com>
16 lines
501 B
C++
16 lines
501 B
C++
#include <torch/csrc/lazy/core/config.h>
|
|
#include <torch/csrc/lazy/ts_backend/config.h>
|
|
|
|
// TODO(whc) unclear if this is useful, has only been tested as true
|
|
C10_DEFINE_bool(
|
|
torch_lazy_ts_tensor_update_sync,
|
|
true,
|
|
"Use synchronous copy inside _copy_from op")
|
|
|
|
// TODO(whc) we need to hook up these flags in a more useful way
|
|
// possibly also keep LTC_TS_CUDA env working?
|
|
C10_DEFINE_bool(
|
|
torch_lazy_ts_cuda,
|
|
false,
|
|
"Use cuda device for torchscript backend (instead of CPU)")
|