We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3a9a94d commit abc2ffbCopy full SHA for abc2ffb
csrc/gpu/oneDNN/Utils.h
@@ -163,7 +163,7 @@ static inline memory::data_type get_onednn_dtype_include_double(
163
const at::Tensor& tensor,
164
bool allow_undef = false) {
165
DeviceId curDevID = at::xpu::current_device();
166
- bool fp64_valid = Settings::I().has_2d_block_array(curDevID);
+ bool fp64_valid = at::xpu::getDeviceProperties(curDevID)->has_fp64;
167
168
if (tensor.scalar_type() == at::ScalarType::Double) {
169
if (fp64_valid)
0 commit comments