From b76b627b61879a15d746c307f28db7a0edd4e4fc Mon Sep 17 00:00:00 2001 From: ziheng Date: Fri, 16 Aug 2019 13:14:29 -0700 Subject: [PATCH] [BUGFIX] Fix for NoneType Target (#3792) --- python/tvm/relay/quantize/_partition.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/tvm/relay/quantize/_partition.py b/python/tvm/relay/quantize/_partition.py index 597c55c..1180d83 100644 --- a/python/tvm/relay/quantize/_partition.py +++ b/python/tvm/relay/quantize/_partition.py @@ -133,7 +133,8 @@ def add_partition_generic(ref_call, new_args, ctx): @register_partition_function("add") def add_partition_function(ref_call, new_args, ctx): """Rewrite function for ewise add for partition""" - if 'cuda' in _target.current_target().keys: + target = _target.current_target() + if target and 'cuda' in target.keys: #TODO(wuwei/ziheng) cuda specific rules return add_partition_generic(ref_call, new_args, ctx) return add_partition_generic(ref_call, new_args, ctx) -- 2.7.4