#pragma once
// TODO: Remove this header
#include <ATen/cuda/DeviceUtils.cuh>