mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
f4ff063c33
# Motivation Add some attributes to `XPUDeviceProp` and expose them via `torch.xpu.get_device_properties` and `torch.xpu.get_device_capability`. They can be used in `torch.compile` or directly passed to triton to generate more optimized code based on device properties. # Additional Context expose the following attributes to `torch.xpu.get_device_properties`： - `has_fp16` (newly added) - `has_fp64` (newly added) - `has_atomic64` (newly added) - `driver_version` - `vendor` - `version` Pull Request resolved: https://github.com/pytorch/pytorch/pull/121898 Approved by: https://github.com/jgong5, https://github.com/EikanWang, https://github.com/malfet, https://github.com/albanD, https://github.com/atalman |
||
|---|---|---|
| .. | ||
| Event.cpp | ||
| Event.h | ||
| Module.cpp | ||
| Module.h | ||
| Stream.cpp | ||
| Stream.h | ||