Skip to content

GitLab

"official/legacy/bert/bert_models.py" did not exist on "3db5347d6561e6fb69bf16c1254e8e48edfc3af2"
Switch branch/tag
Find file BlameHistoryPermalink
  • Paul Fultz II's avatar
    Conditionally enable pointwise fusion (#992) · 157935ff
    Paul Fultz II authored 
    This enables the pointwise fusions using the MIGRAPHX_ENABLE_POINTWISE_FUSION env variable. Its disabled by default since MIOpen fusions need to be refactored.

This also adds a compile_ops pass to compile the pointwise modules. All tests except test_gpu_fast_math passes with MIGRAPHX_ENABLE_POINTWISE_FUSION=1 set.
    157935ff
fuse_pointwise.cpp 5.36 KB