Skip to content

Commit abc2ffb

Browse files
xiaolil1gujinghui
andauthored
fix onednn double check issue (#4688) (#4731)
Co-authored-by: Jinghui <jinghui.gu@intel.com>
1 parent 3a9a94d commit abc2ffb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

csrc/gpu/oneDNN/Utils.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ static inline memory::data_type get_onednn_dtype_include_double(
163163
const at::Tensor& tensor,
164164
bool allow_undef = false) {
165165
DeviceId curDevID = at::xpu::current_device();
166-
bool fp64_valid = Settings::I().has_2d_block_array(curDevID);
166+
bool fp64_valid = at::xpu::getDeviceProperties(curDevID)->has_fp64;
167167

168168
if (tensor.scalar_type() == at::ScalarType::Double) {
169169
if (fp64_valid)

0 commit comments

Comments
 (0)