Skip to content

Commit cf99d85

Browse files
AutoCast: make fall through for maxpool2d and adaptive_avg_pool2d (#4)
1 parent 53c605a commit cf99d85

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torch_ipex/csrc/autocast_kernel.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ at::Tensor max_pool2d(const at::Tensor& input, at::IntArrayRef kernel_size, at::
137137
#if defined(ENABLE_AUTOCAST_VERBOSE)
138138
verbose::OpNameGuard op_name("max_pool2d");
139139
#endif
140-
return at::max_pool2d(cpu_cached_cast(at::kFloat, input), kernel_size, stride, padding, dilation, ceil_mode);
140+
return at::max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode);
141141
}
142142

143143
at::Tensor adaptive_avg_pool2d(const at::Tensor& input, at::IntArrayRef output_size) {
@@ -150,7 +150,7 @@ at::Tensor adaptive_avg_pool2d(const at::Tensor& input, at::IntArrayRef output_s
150150
#if defined(ENABLE_AUTOCAST_VERBOSE)
151151
verbose::OpNameGuard op_name("adaptive_avg_pool2d");
152152
#endif
153-
return at::adaptive_avg_pool2d(cpu_cached_cast(at::kFloat, input), output_size);
153+
return at::adaptive_avg_pool2d(input, output_size);
154154
}
155155

156156
at::Tensor relu(const at::Tensor& input) {

0 commit comments

Comments
 (0)