1*da0073e9SAndroid Build Coastguard Worker #pragma once 2*da0073e9SAndroid Build Coastguard Worker // TODO: Remove once torchvision has been updated to use the ATen header 3*da0073e9SAndroid Build Coastguard Worker #include <ATen/cuda/Atomic.cuh> 4