Safe Haskell	None
Language	Haskell2010

Torch.Device

Documentation

data DeviceType Source #

Constructors

CPU
CUDA
MPS

Instances

Instances details

Show DeviceType Source #
Instance details Defined in Torch.Device Methods showsPrec :: Int -> DeviceType -> ShowS # show :: DeviceType -> String # showList :: [DeviceType] -> ShowS #
Eq DeviceType Source #
Instance details Defined in Torch.Device Methods (==) :: DeviceType -> DeviceType -> Bool # (/=) :: DeviceType -> DeviceType -> Bool #
Ord DeviceType Source #
Instance details Defined in Torch.Device Methods compare :: DeviceType -> DeviceType -> Ordering # (<) :: DeviceType -> DeviceType -> Bool # (<=) :: DeviceType -> DeviceType -> Bool # (>) :: DeviceType -> DeviceType -> Bool # (>=) :: DeviceType -> DeviceType -> Bool # max :: DeviceType -> DeviceType -> DeviceType # min :: DeviceType -> DeviceType -> DeviceType #
Castable DeviceType DeviceType Source #
Instance details Defined in Torch.Device Methods cast :: DeviceType -> (DeviceType -> IO r) -> IO r # uncast :: DeviceType -> (DeviceType -> IO r) -> IO r #
HasToDevice device' device (HList xs) (HList ys) => ReduceGradients device' '[device] ('[HList xs] :: [Type]) (ys :: [k]) Source #
Instance details Defined in Torch.Typed.NN.DataParallel Methods reduceGradients :: HList '[HList xs] -> HList ys Source #
(HasToDevice device' device (HList xs) (HList ys), ReduceGradients device' devices xxs ys, HZipWith SumF ys ys ys, 1 <= ListLength xxs) => ReduceGradients device' (device ': devices) (HList xs ': xxs :: [Type]) (ys :: [k]) Source #
Instance details Defined in Torch.Typed.NN.DataParallel Methods reduceGradients :: HList (HList xs ': xxs) -> HList ys Source #
HasReplicate ('[] :: [(DeviceType, Nat)]) device f ('[] :: [Type]) Source #
Instance details Defined in Torch.Typed.Device Methods replicate :: f -> HList ('[] :: [Type]) Source #
HasToDevices ('[] :: [(DeviceType, Nat)]) ('[] :: [(DeviceType, Nat)]) ('[] :: [Type]) ('[] :: [Type]) Source #
Instance details Defined in Torch.Typed.Device Methods toDevices :: HList ('[] :: [Type]) -> HList ('[] :: [Type]) Source #
(chunks ~ ListLength devices', tensorChunks ~ (Chunk chunks 0 shape dtype device :: [Type]), Castable (HList tensorChunks) [ATenTensor], devices ~ HReplicateR chunks device, HasToDevices devices' devices tensorChunks gs, KnownNat chunks) => HasScatter (devices' :: [(DeviceType, Nat)]) (device :: (DeviceType, Nat)) (Tensor device dtype shape) (gs :: [Type]) Source #
Instance details Defined in Torch.Typed.Device Methods scatter :: Tensor device dtype shape -> HList gs Source #
(HZipWithM Concurrently GradConcurrentlyF parameters losses gradients', ReduceGradients device' devices gradients' gradients) => HasGradConcurrently (device' :: (DeviceType, Nat)) (devices :: [(DeviceType, Nat)]) (parameters :: [k1]) (losses :: [k1]) (gradients :: [k2]) Source #
Instance details Defined in Torch.Typed.NN.DataParallel Methods gradConcurrently :: HList parameters -> HList losses -> Concurrently (HList gradients) Source #
(chunks ~ ListLength fs, devices ~ GetDevices fs, devices' ~ HReplicateR chunks device', HasToDevices devices' devices fs tensorChunks, '(shape, dtype, device') ~ Cat 0 tensorChunks, Castable (HList tensorChunks) [ATenTensor]) => HasGather (device' :: (DeviceType, Nat)) (devices :: [(DeviceType, Nat)]) (fs :: [Type]) (Tensor device' dtype shape) Source #
Instance details Defined in Torch.Typed.Device Methods gather :: HList fs -> Tensor device' dtype shape Source #
(HasReplicate devices' device f gs, HasToDevice device' device f g) => HasReplicate (device' ': devices') device f (g ': gs) Source #
Instance details Defined in Torch.Typed.Device Methods replicate :: f -> HList (g ': gs) Source #
(HasToDevices devices' devices fs gs, HasToDevice device' device f g) => HasToDevices (device' ': devices') (device ': devices) (f ': fs) (g ': gs) Source #
Instance details Defined in Torch.Typed.Device Methods toDevices :: HList (f ': fs) -> HList (g ': gs) Source #
KnownNat n => KnownDevice '('CPU, n) Source #
Instance details Defined in Torch.Typed.Tensor Methods deviceVal :: Device Source #
KnownNat n => KnownDevice '('CUDA, n) Source #
Instance details Defined in Torch.Typed.Tensor Methods deviceVal :: Device Source #
KnownNat n => KnownDevice '('MPS, n) Source #
Instance details Defined in Torch.Typed.Tensor Methods deviceVal :: Device Source #

data Device Source #

Constructors

Device
Fields deviceType :: DeviceType deviceIndex :: Int16

Instances

Instances details

Show Device Source #
Instance details Defined in Torch.Device Methods showsPrec :: Int -> Device -> ShowS # show :: Device -> String # showList :: [Device] -> ShowS #
Eq Device Source #
Instance details Defined in Torch.Device Methods (==) :: Device -> Device -> Bool # (/=) :: Device -> Device -> Bool #
Ord Device Source #
Instance details Defined in Torch.Device Methods compare :: Device -> Device -> Ordering # (<) :: Device -> Device -> Bool # (<=) :: Device -> Device -> Bool # (>) :: Device -> Device -> Bool # (>=) :: Device -> Device -> Bool # max :: Device -> Device -> Device # min :: Device -> Device -> Device #
Show (TVar (Either (Word64, Device) (ForeignPtr Generator))) Source #
Instance details Defined in Torch.Random Methods showsPrec :: Int -> TVar (Either (Word64, Device) (ForeignPtr Generator)) -> ShowS # show :: TVar (Either (Word64, Device) (ForeignPtr Generator)) -> String # showList :: [TVar (Either (Word64, Device) (ForeignPtr Generator))] -> ShowS #