search

CoinCheung / pytorch-loss

label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful
MIT License
2.17k stars 374 forks source link

remove THC.h #37

Closed zyddnys closed 2 years ago

zyddnys commented 2 years ago

Latest pytorch (1.11) has removed THC.h, with that gone the following changes were also made so it can be compiled:

  1. THCudaCheck -> AT_CUDA_CHECK
  2. THCCeilDiv was reintroduced into common.hpp
  3. common.hpp were added to all cuda source files.
CoinCheung commented 2 years ago

Thanks for updating this !!!