From 477a67731736315a8c87cb60780a95d6f2985eec Mon Sep 17 00:00:00 2001
From: Alexander Zinoviev <8257131+alexander-zinoviev@users.noreply.github.com>
Date: Mon, 7 Oct 2024 09:39:11 -0700
Subject: [PATCH] Fix typos in test/unit/conv/cache_testbed_output.h (#1652)

Co-authored-by: Alexander Zinoviev <azinoviev@tesla.com>
---
 test/unit/conv/cache_testbed_output.h | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/test/unit/conv/cache_testbed_output.h b/test/unit/conv/cache_testbed_output.h
index 4f3981e83..eec63a797 100644
--- a/test/unit/conv/cache_testbed_output.h
+++ b/test/unit/conv/cache_testbed_output.h
@@ -659,7 +659,7 @@ inline CachedTestKey CreateCachedConv2dTestKey(
   ElementCompute alpha,
   ElementCompute beta,
   cutlass::TensorView<ElementA, LayoutA> A,
-  cutlass::TensorView<ElementA, LayoutB> B,
+  cutlass::TensorView<ElementB, LayoutB> B,
   cutlass::TensorView<ElementC, LayoutC> C
 ) {
 
@@ -711,7 +711,7 @@ inline CachedTestKey CreateCachedConv2dWithBroadcastTestKey(
   ElementCompute alpha,
   ElementCompute beta,
   cutlass::TensorView<ElementA, LayoutA> A,
-  cutlass::TensorView<ElementA, LayoutB> B,
+  cutlass::TensorView<ElementB, LayoutB> B,
   cutlass::TensorView<ElementC, LayoutC> C
 ) {
 
@@ -763,7 +763,7 @@ inline CachedTestKey CreateCachedConv2dWithReductionTestKey(
   ElementCompute alpha,
   ElementCompute beta,
   cutlass::TensorView<ElementA, LayoutA> A,
-  cutlass::TensorView<ElementA, LayoutB> B,
+  cutlass::TensorView<ElementB, LayoutB> B,
   cutlass::TensorView<ElementC, LayoutC> C
 ) {
 
@@ -814,7 +814,7 @@ inline CachedTestKey CreateCachedConv3dTestKey(
   ElementCompute alpha,
   ElementCompute beta,
   cutlass::TensorView<ElementA, LayoutA> A,
-  cutlass::TensorView<ElementA, LayoutB> B,
+  cutlass::TensorView<ElementB, LayoutB> B,
   cutlass::TensorView<ElementC, LayoutC> C
 ) {