Import the implementation from Thrust