Skip to content

Commit 6ccf197

Browse files
razarmehrkulinseth
authored andcommitted
Fix correctness issues with Upsample 1D and 2D (#183)
* Fix correctness issues with Upsample 1D and 2D - Implemented following new ops: upsample_nearest1d_backward upsample_nearest_exact1d upsample_nearest_exact1d_backward - Moved Upsample code from Shape.mm to Upsample.mm * Corrections on some comments * Fix the leak in cat_out_mps and clean up
1 parent d84cf3d commit 6ccf197

File tree

5 files changed

+456
-390
lines changed

5 files changed

+456
-390
lines changed

aten/src/ATen/mps/MPSFallback.mm

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ Tensor slow_conv2d_forward_mps(
6262
m.impl("linalg_vector_norm", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
6363
m.impl("sgn.out", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
6464
m.impl("_slow_conv2d_forward", slow_conv2d_forward_mps);
65+
m.impl("upsample_nearest3d.vec", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
6566
}
6667

6768
} // namespace at

0 commit comments

Comments
 (0)