This PR fixes the error when building the emd_kernel extension with some older versions of Pytorch. The problem is caused by the previous PR #58, which change CHECK_EQ to the new TORCH_CHECK_EQ macro. However, TORCH_CHECK_EQ is only introduced in the newer version of Pytorch. See Pytorch PR 82032
This PR fixes the error when building the emd_kernel extension with some older versions of Pytorch. The problem is caused by the previous PR #58, which change CHECK_EQ to the new TORCH_CHECK_EQ macro. However, TORCH_CHECK_EQ is only introduced in the newer version of Pytorch. See Pytorch PR 82032