Skip to content

Commit 616a6d3

Browse files
authored
Update Gluon pipeline (#4696)
Added SCCP and resolve_auto_configs to the Gluon pipeline, and renamed the pipeline to follow upstream code. Signed-off-by: Tiotto, Ettore <[email protected]>
1 parent cb24881 commit 616a6d3

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

third_party/intel/backend/compiler.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -312,18 +312,20 @@ def make_ttgir(mod, metadata, opt, properties):
312312
metadata["cluster_dims"] = (cluster_info.clusterDimX, cluster_info.clusterDimY, cluster_info.clusterDimZ)
313313
return mod
314314

315-
@staticmethod
316-
def ttgir_opt(src, metadata, options):
315+
def gluon_to_ttgir(self, src, metadata, options):
317316
mod = src
318317
pm = ir.pass_manager(mod.context)
319318
pm.enable_debug()
320319

321320
passes.ttgpuir.add_inliner(pm)
321+
passes.gluon.add_resolve_auto_encodings(pm)
322+
passes.common.add_sccp(pm)
322323
passes.ttir.add_loop_aware_cse(pm)
323324
passes.ttgpuir.add_canonicalizer(pm)
324325
passes.ttgpuir.add_combine_tensor_select_and_if(pm)
325326

326327
pm.run(mod)
328+
metadata["tensordesc_meta"] = mod.get_tensordesc_metadata()
327329
return mod
328330

329331
@staticmethod
@@ -452,7 +454,7 @@ def add_stages(self, stages, options, language):
452454
stages["ttir"] = lambda src, metadata: self.make_ttir(src, metadata, options)
453455
stages["ttgir"] = lambda src, metadata: self.make_ttgir(src, metadata, options, self.properties)
454456
elif language == Language.GLUON:
455-
stages["ttgir"] = lambda src, metadata: self.ttgir_opt(src, metadata, options)
457+
stages["ttgir"] = lambda src, metadata: self.gluon_to_ttgir(src, metadata, options)
456458
stages["llir"] = lambda src, metadata: self.make_llir(src, metadata, options)
457459
stages["spv"] = lambda src, metadata: self.make_spv(src, metadata, options, self.device_arch)
458460

0 commit comments

Comments
 (0)