Unverified Commit 4a694cc0 authored by lml131's avatar lml131 Committed by GitHub
Browse files

add PARROTS_JIT_OPTION (#816)

* code update

* add PARROTS_JIT_OPTION

* rm flow_warp_module

* fix for lint

* small fix
parent 5df759ad
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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment