pull/3219/head
YH 2 years ago committed by GitHub
parent 933048ad3e
commit 8f740deb53
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -168,12 +168,12 @@ def _get_grad_args(*args):
# if there is no grad tensor, the backward of PreFwdPostBwd can't be triggered
arg_zero = args[0]
if not isinstance(arg_zero, tuple):
raise NotImplementedError("Some torch function is incompatible because of its complcated inputs.")
raise NotImplementedError("Some torch function is incompatible because of its complicated inputs.")
check_grad_flag = False
for obj in arg_zero:
check_grad_flag |= _is_grad_tensor(obj)
if not check_grad_flag:
raise NotImplementedError("Some torch function is incompatible because of its complcated inputs.")
raise NotImplementedError("Some torch function is incompatible because of its complicated inputs.")
return arg_zero, args[1:]

Loading…
Cancel
Save