Eddie Van Halen said that 'every riff has been done twice' and Steve Earle said that 'when you aren't feeling totally original, it's ok to creatively graft from others so long is it becomes your own.

So...if you steal a whole riff, then not cool. If a riff inspires you to build from that vibe, why not?