print-diagnostics throws a warning instead of getting terminated when encountering modules do not require gradient

print-diagnostics throws a warning instead of getting terminated when encountering modules do not require gradient #342