1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
#include <torch/csrc/lazy/core/config.h>
// TODO(whc) unclear if this is useful, has only been tested as true
// NOLINTNEXTLINE(misc-use-internal-linkage)
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?
// NOLINTNEXTLINE(misc-use-internal-linkage)
C10_DEFINE_bool(
torch_lazy_ts_cuda,
false,
"Use cuda device for torchscript backend (instead of CPU)")
|
