Call narrow only for TensorCoreTiledLayout (#1207)
* Call narrow only for TensorCoreTiledLayout only Summary: att, previously in https://github.com/pytorch/ao/pull/914 we added narrow op for all layout, the introduced narrow op breaks the pattern for int8 dynamic activation int4 weight quant for executorch, this PR guarded narrow op for tensor core tiled layout only If similar things coming up in the future we can factor this into a proper API for Layout or TensorImpl Test Plan: python test/test_integration.py -k test_export Reviewers: Subscribers: Tasks: Tags: * enable test * version * skip aoti * version update * skip aoti
J
Jerry Zhang committed
ccd883b9344fa7d8d0e4637f50372ba07ef5f0fa
Parent: 2ba1a61
Committed by GitHub <noreply@github.com>
on 11/12/2024, 6:05:05 AM