Skip to content

Commit 2538821

Browse files
committed
Remove cluster_dims references from Intel's backend to match upstream
1 parent 51245e9 commit 2538821

File tree

2 files changed

+0
-17
lines changed

2 files changed

+0
-17
lines changed

third_party/intel/backend/compiler.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -246,12 +246,6 @@ def make_ttir(cls, mod, metadata, opt):
246246
@classmethod
247247
@track
248248
def make_ttgir(cls, mod, metadata, opt, properties):
249-
cluster_info = intel.ClusterInfo()
250-
if opt.cluster_dims is not None:
251-
cluster_info.clusterDimX = opt.cluster_dims[0]
252-
cluster_info.clusterDimY = opt.cluster_dims[1]
253-
cluster_info.clusterDimZ = opt.cluster_dims[2]
254-
255249
# Annotate module with information required by subsequent transformations.
256250
pm = ir.pass_manager(mod.context)
257251
pm.enable_debug()
@@ -303,7 +297,6 @@ def make_ttgir(cls, mod, metadata, opt, properties):
303297
intel.passes.ttgpuir.add_optimize_reduction_locality(pm)
304298
intel.passes.arith.add_arith_emulate_unsupported_floats(pm, ["bf16"], "f32")
305299
pm.run(mod, 'make_ttgir')
306-
metadata["cluster_dims"] = (cluster_info.clusterDimX, cluster_info.clusterDimY, cluster_info.clusterDimZ)
307300
return mod
308301

309302
def gluon_to_ttgir(self, src, metadata, options):

third_party/intel/backend/driver.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -738,16 +738,6 @@ def format_of(ty):
738738
int threads_per_warp = PyLong_AsLong(threads_per_warp_attr);
739739
Py_DECREF(threads_per_warp_attr);
740740
741-
// extract cluster dims
742-
PyObject *clusterDim = PyObject_GetAttrString(kernel_metadata, "cluster_dims");
743-
if (!PyTuple_Check(kernel_metadata)) {{
744-
PyErr_SetString(PyExc_TypeError, "kernel_metadata.cluster_dims must be a tuple");
745-
return NULL;
746-
}}
747-
int clusterDimX = PyLong_AsLong(PyTuple_GetItem(clusterDim, 0));
748-
int clusterDimY = PyLong_AsLong(PyTuple_GetItem(clusterDim, 1));
749-
int clusterDimZ = PyLong_AsLong(PyTuple_GetItem(clusterDim, 2));
750-
Py_DECREF(clusterDim);
751741
// extract launch metadata
752742
if (launch_enter_hook != Py_None){{
753743
PyObject* ret = PyObject_CallOneArg(launch_enter_hook, launch_metadata);

0 commit comments

Comments
 (0)