mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
6ddaf3bd37
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68154 Test Plan: added a basic test; cover more by using lazy_tensor_staging tests Reviewed By: Krovatkin, alanwaketan Differential Revision: D32224303 fbshipit-source-id: ac3e1161229b8ae60fdb15ffa72e17072b595914
8 lines
240 B
C++
8 lines
240 B
C++
#include <torch/csrc/lazy/core/config.h>
|
|
|
|
// TODO(whc) either deprecate this, or use it for all shape inference
|
|
C10_DEFINE_int(
|
|
torch_lazy_ts_shape_cache_size,
|
|
4096,
|
|
"Set the size for the shape cache used for shape inference");
|