3-D heat equation solved by finite differences using Thrust