1 | #include <torch/csrc/lazy/core/config.h> |
---|---|
2 | |
3 | // TODO(whc) unclear if this is useful, has only been tested as true |
4 | C10_DEFINE_bool( |
5 | torch_lazy_ts_tensor_update_sync, |
6 | true, |
7 | "Use synchronous copy inside _copy_from op"); |
8 | |
9 | // TODO(whc) we need to hook up these flags in a more useful way |
10 | // possibly also keep LTC_TS_CUDA env working? |
11 | C10_DEFINE_bool( |
12 | torch_lazy_ts_cuda, |
13 | false, |
14 | "Use cuda device for torchscript backend (instead of CPU)"); |
15 |