import paddle in primops.py

0x45f · May 11, 2022 · 8a1e8e9 · 8a1e8e9
1 parent 2875959
commit 8a1e8e9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/paddle/incubate/autograd/primops.py b/python/paddle/incubate/autograd/primops.py
@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import paddle
 from paddle.fluid.layer_helper import LayerHelper
 from .primreg import REGISTER_FN