fix

segyges · Apr 19, 2023 · b893d03 · b893d03
1 parent a766a1b
commit b893d03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/instance_norm_nvfuser_kernel.cu b/csrc/instance_norm_nvfuser_kernel.cu
@@ -159,7 +159,7 @@ std::vector<at::Tensor> instance_norm_nvfuser_backward(
   ) {
     InstanceNormKey backward_key;
     memset(&backward_key, 0, sizeof(InstanceNormKey));
-    getKey(input, weight, run_mean, channels_last, eps, backward_key);
+    setKey(input, weight, run_mean, channels_last, eps, backward_key);
     if (backward_fusion_cache.find(backward_key) == backward_fusion_cache.end()) {
       auto fusion = std::make_unique<Fusion>();
       FusionGuard fg(fusion.get());