Skip to content

Commit

Permalink
Merge pull request #52 from bsteinb/main
Browse files Browse the repository at this point in the history
Add Ada and Hopper to DeviceArchitecture
  • Loading branch information
Cldfire authored Feb 10, 2024
2 parents 1d71e81 + 2574e2d commit 26d78ad
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 0 deletions.
4 changes: 4 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@ Updates for NVML 12.2.
* `Device`
* Methods
* `pcie_link_speed()`
* `DeviceArchitecture`
* Variants
* `Ada`
* `Hopper`

### Changed

Expand Down
10 changes: 10 additions & 0 deletions nvml-wrapper/src/enums/device.rs
Original file line number Diff line number Diff line change
Expand Up @@ -221,6 +221,10 @@ pub enum DeviceArchitecture {
Turing,
/// <https://en.wikipedia.org/wiki/Ampere_(microarchitecture)>
Ampere,
/// <https://en.wikipedia.org/wiki/Ada_Lovelace_(microarchitecture)>
Ada,
/// <https://en.wikipedia.org/wiki/Hopper_(microarchitecture)>
Hopper,
/// Unknown device architecture (most likely something newer).
Unknown,
}
Expand All @@ -235,6 +239,8 @@ impl DeviceArchitecture {
Self::Volta => NVML_DEVICE_ARCH_VOLTA,
Self::Turing => NVML_DEVICE_ARCH_TURING,
Self::Ampere => NVML_DEVICE_ARCH_AMPERE,
Self::Ada => NVML_DEVICE_ARCH_ADA,
Self::Hopper => NVML_DEVICE_ARCH_HOPPER,
Self::Unknown => NVML_DEVICE_ARCH_UNKNOWN,
}
}
Expand All @@ -251,6 +257,8 @@ impl TryFrom<nvmlDeviceArchitecture_t> for DeviceArchitecture {
NVML_DEVICE_ARCH_VOLTA => Ok(Self::Volta),
NVML_DEVICE_ARCH_TURING => Ok(Self::Turing),
NVML_DEVICE_ARCH_AMPERE => Ok(Self::Ampere),
NVML_DEVICE_ARCH_ADA => Ok(Self::Ada),
NVML_DEVICE_ARCH_HOPPER => Ok(Self::Hopper),
NVML_DEVICE_ARCH_UNKNOWN => Ok(Self::Unknown),
_ => Err(NvmlError::UnexpectedVariant(data)),
}
Expand All @@ -266,6 +274,8 @@ impl Display for DeviceArchitecture {
Self::Volta => f.write_str("Volta"),
Self::Turing => f.write_str("Turing"),
Self::Ampere => f.write_str("Ampere"),
Self::Ada => f.write_str("Ada"),
Self::Hopper => f.write_str("Hopper"),
Self::Unknown => f.write_str("Unknown"),
}
}
Expand Down

0 comments on commit 26d78ad

Please sign in to comment.