Skip to content

Support GPU/CUDA inference in torchscript model; fix an issue

Compare
Choose a tag to compare
@taoleicn taoleicn released this 13 May 17:22
· 10 commits to torchscript_gpu_v2.5 since this release

Dev1:

  • Support GPU/CUDA inference in torchscript model
  • Support post layer norm
  • Support custom init value for weight_c
    Dev2:
  • Fix an issue