From 4a694cc0eeedf4315ff020df23f9aa7530030b5a Mon Sep 17 00:00:00 2001
From: lml131 <43259216+lml131@users.noreply.github.com>
Date: Fri, 5 Feb 2021 19:58:33 +0800
Subject: [PATCH] add PARROTS_JIT_OPTION (#816)

* code update

* add PARROTS_JIT_OPTION

* rm flow_warp_module

* fix for lint

* small fix
---
 mmcv/utils/parrots_jit.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/mmcv/utils/parrots_jit.py b/mmcv/utils/parrots_jit.py
index 609c11acd0..440d4bcbd2 100644
--- a/mmcv/utils/parrots_jit.py
+++ b/mmcv/utils/parrots_jit.py
@@ -1,6 +1,10 @@
+import os
+
 from .parrots_wrapper import TORCH_VERSION
 
-if TORCH_VERSION == 'parrots':
+parrots_jit_option = os.getenv('PARROTS_JIT_OPTION')
+
+if TORCH_VERSION == 'parrots' and parrots_jit_option == 'ON':
     from parrots.jit import pat as jit
 else: