mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
40fb738197
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/140236 Approved by: https://github.com/ezyang
17 lines
544 B
C++
17 lines
544 B
C++
#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)")
|