rocmPackages.miopen: add impureTest for droupout kernels

2026-04-07 14:56:45 -07:00
parent 853d5aa780
commit 61efb76ea1
1 changed files with 5 additions and 0 deletions
@@ -295,6 +295,11 @@ stdenv.mkDerivation (finalAttrs: {
      name = "conv";
      testScript = "MIOpenDriver conv -n 1 -c 1 -H 4 -W 4 -k 1 -y 3 -x 3 -p 0 -q 0 -V 0";
    };
+    dropout = callPackage ./test-runtime-compilation.nix {
+      miopen = finalAttrs.finalPackage;
+      name = "dropout";
+      testScript = "MIOpenDriver dropout -d 64,32,14,14";
+    };
    pool = callPackage ./test-runtime-compilation.nix {
      miopen = finalAttrs.finalPackage;
      name = "pool";