Skip to content

Commit 46ac8dd

Browse files
committed
Check opencl extensions on LTS2 driver
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 386e419 commit 46ac8dd

File tree

1 file changed

+4
-10
lines changed

1 file changed

+4
-10
lines changed

third_party/intel/backend/driver.py

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -931,18 +931,12 @@ def update_advanced_features(device, dev_property):
931931
dev_property["has_bfloat16_conversions"] = "cl_intel_bfloat16_conversions" in supported_extensions
932932
else:
933933
check = self.utils.has_opencl_extension
934-
# FIXME: eventually even LTS driver will support OpenCL extensions.
935-
# Please remove this after upgrading to a new version.
936-
# https://github.com/intel/intel-xpu-backend-for-triton/issues/4708
937-
is_lts = "1.3" in dev_property["driver_version"]
938934
dev_property["has_subgroup_matrix_multiply_accumulate"] = check(
939-
device, b"cl_intel_subgroup_matrix_multiply_accumulate") if not is_lts else False
935+
device, b"cl_intel_subgroup_matrix_multiply_accumulate")
940936
dev_property["has_subgroup_matrix_multiply_accumulate_tensor_float32"] = check(
941-
device, b"cl_intel_subgroup_matrix_multiply_accumulate_tensor_float32") if not is_lts else False
942-
dev_property["has_subgroup_2d_block_io"] = check(
943-
device, b"cl_intel_subgroup_2d_block_io") if not is_lts else False
944-
dev_property["has_bfloat16_conversions"] = check(
945-
device, b"cl_intel_bfloat16_conversions") if not is_lts else False
937+
device, b"cl_intel_subgroup_matrix_multiply_accumulate_tensor_float32")
938+
dev_property["has_subgroup_2d_block_io"] = check(device, b"cl_intel_subgroup_2d_block_io")
939+
dev_property["has_bfloat16_conversions"] = check(device, b"cl_intel_bfloat16_conversions")
946940

947941
update_advanced_features(device, dev_property)
948942
return GPUTarget("xpu", dev_property, warp_size=32)

0 commit comments

Comments
 (0)