Skip to content

PyTorch/CSPRNG now supports torch.randperm

Compare
Choose a tag to compare
@pbelevich pbelevich released this 31 Aug 20:06
dc42a5c

PyTorch/CSPRNG 0.1.2 Release Notes

Features

The following list of methods are now supported:

Kernel CUDA CPU
randperm(n) yes* yes
  • the calculations are done on CPU and the result is copied to CUDA

Bug Fixes

Strict-aliasing violation in csprng.h fixed (#68)