commit 9e2c74cc585819a671cb9bd7c66b23514b97af48
author Huazhong Ning <ninghz@fb.com> Sat Jul 15 19:19:57 2017 -0700
committer Facebook Github Bot <facebook-github-bot@users.noreply.github.com> Sat Jul 15 19:22:32 2017 -0700
tree 74a81a0c51e49b8299c2febc719b601576e1669b
parent b6691277f5fa74c5edd69cb773e453551c2322ee

Use scope name for dataset cursor

Summary: Currently the dataset cursor blob is using a fixed name. When we read from multi input tables, the dataset cursor of each table is using the same blob. This messed up the split queue and crashed the reader pipelines (see the errors and failures in https://fb.quip.com/uzbIA7K0PgVe)

Reviewed By: dragonxlwang, rayleichen

Differential Revision: D5419863

fbshipit-source-id: 5983a3d8d2e286dc47c2ec38ed1dbbe30c7c9b49

caffe2/python/dataset.py

diff --git a/caffe2/python/dataset.py b/caffe2/python/dataset.py
index bb8abfa..73b88b5 100644
--- a/caffe2/python/dataset.py
+++ b/caffe2/python/dataset.py
@@ -35,7 +35,7 @@
         if self.cursor is None:
             self.cursor = init_net.CreateTreeCursor(
                 [],
-                [self.name],
+                init_net.NextScopedBlob(self.name),
                 fields=self.dataset.fields)
 
     def read(self, read_net):