diff --git a/mlir-requirements.txt b/mlir-requirements.txt index 2e3f001355f..02af8d38df6 100644 --- a/mlir-requirements.txt +++ b/mlir-requirements.txt @@ -21,4 +21,4 @@ # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN # THE SOFTWARE. ##################################################################################### -ROCmSoftwarePlatform/rocMLIR@3657f509bfed86bb79d5c6e24aa237e48f09f9f3 -DBUILD_FAT_LIBROCKCOMPILER=On +ROCmSoftwarePlatform/rocMLIR@2c519c48eaa278d13e6c40bc0941119826d71512 -DBUILD_FAT_LIBROCKCOMPILER=On diff --git a/src/targets/gpu/fuse_mlir.cpp b/src/targets/gpu/fuse_mlir.cpp index 33d88ff5a6b..bdd7cbd2f65 100644 --- a/src/targets/gpu/fuse_mlir.cpp +++ b/src/targets/gpu/fuse_mlir.cpp @@ -235,8 +235,7 @@ struct find_mlir_fused_ops "log", "recip", "rsqrt", - // There are bugs in MLIR right now for models using sigmoid so disable it for now - // "sigmoid", + "sigmoid", "softmax", "tanh", };