Skip to content

Commit

Permalink
Mark IPU device as not supports_as_strided (#89130)
Browse files Browse the repository at this point in the history
Currently causes issues in calls to `.to`.
Pull Request resolved: #89130
Approved by: https://github.com/albanD
  • Loading branch information
charlie-wt committed Dec 1, 2022
1 parent ae2fe40 commit 5bf9a55
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 5bf9a55

Please sign in to comment.