Skip to content

Commit

Permalink
Mark IPU device as not supports_as_strided
Browse files Browse the repository at this point in the history
  • Loading branch information
charlie-wt committed Nov 16, 2022
1 parent 5270122 commit 3305707
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion c10/core/Device.h
Expand Up @@ -148,7 +148,8 @@ struct C10_API Device final {

/// Return true if the device supports arbirtary strides.
bool supports_as_strided() const noexcept {
return type_ != DeviceType::XLA && type_ != DeviceType::Lazy;
return type_ != DeviceType::IPU && type_ != DeviceType::XLA &&
type_ != DeviceType::Lazy;
}

/// Same string as returned from operator<<.
Expand Down

0 comments on commit 3305707

Please sign in to comment.