diff options
Diffstat (limited to 'caffe2/python/pybind_state.cc')
| -rw-r--r-- | caffe2/python/pybind_state.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/caffe2/python/pybind_state.cc b/caffe2/python/pybind_state.cc index 32e0a86603..7f0b0c81d8 100644 --- a/caffe2/python/pybind_state.cc +++ b/caffe2/python/pybind_state.cc @@ -55,6 +55,10 @@ CAFFE_DEFINE_TYPED_REGISTRY( REGISTER_BLOB_FETCHER((TypeMeta::Id<TensorCPU>()), TensorFetcher<CPUContext>); REGISTER_BLOB_FEEDER(CPU, TensorFeeder<CPUContext>); +Workspace* GetCurrentWorkspace() { + return gWorkspace; +} + class StringFetcher : public BlobFetcherBase { public: py::object Fetch(const Blob& blob) override { |