libtorch-ffi-1.11.0.0: test out alternative options for ffi interface to libtorch 1.x
Safe HaskellSafe-Inferred
LanguageHaskell2010

Torch.Internal.Managed.Type.Generator

Documentation

newCUDAGenerator :: Word16 -> IO (ForeignPtr Generator) Source #

newCPUGenerator :: Word64 -> IO (ForeignPtr Generator) Source #

generator_set_current_seed :: ForeignPtr Generator -> Word64 -> IO () Source #

generator_current_seed :: ForeignPtr Generator -> IO Word64 Source #

generator_seed :: ForeignPtr Generator -> IO Word64 Source #

generator_clone :: ForeignPtr Generator -> IO (ForeignPtr Generator) Source #

generator_get_device :: ForeignPtr Generator -> IO Int64 Source #

generator_is_cpu :: ForeignPtr Generator -> IO CBool Source #

generator_is_cuda :: ForeignPtr Generator -> IO CBool Source #

generator_is_hip :: ForeignPtr Generator -> IO CBool Source #