Snakepit.Error.DeviceMismatch exception (Snakepit v0.8.7)

View Source

Device mismatch error for tensor operations.

Raised when tensors on different devices are used in an operation that requires them to be on the same device.

Summary

Types

device()
t()

Types

device()

@type device() ::
  :cpu | :mps | {:cuda, non_neg_integer()} | {:rocm, non_neg_integer()}

t()

@type t() :: %Snakepit.Error.DeviceMismatch{
  __exception__: true,
  expected: device() | nil,
  got: device() | nil,
  message: String.t(),
  operation: String.t() | nil
}