handle memset_pattern16

EnzymeAD · Dec 10, 2023 · 607a43c · 607a43c
1 parent 4ee70bd
commit 607a43c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/WORKSPACE b/WORKSPACE
@@ -61,7 +61,7 @@ load("@rules_python//python/pip_install:repositories.bzl", "pip_install_dependen
 pip_install_dependencies()
 
 ENZYME_COMMIT = "cbb970161fd41ce55da028f0960a441382b07112"
-ENZYME_SHA256 = ""
+ENZYME_SHA256 = "ec0450fdbc7f18cab46492acd3288b8347fa222317f9ff475768f5f10c45478c"
 
 http_archive(
     name = "enzyme",

diff --git a/enzyme_jax/clang_compile.cc b/enzyme_jax/clang_compile.cc
@@ -612,6 +612,10 @@ struct tensor<T, n0, N...>
           if (MTI->getSource() == prev)
             continue;
         }
+        if (auto CI = dyn_cast<CallInst>(cur))
+          if (auto F = CI->getCalledFunction())
+            if (F->getName() == "memset_pattern16")
+              continue;
         if (auto MS = dyn_cast<MemSetInst>(cur)) {
           toErase.insert(MS);
           continue;