Skip to content

Commit 0d58893

Browse files
Add to_plain_if_needed for upsample nearest (#1641)
1 parent 7ee2fbf commit 0d58893

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

scripts/tools/codegen/dest/register_dispatch_key.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -264,10 +264,10 @@ class RegisterDispatchKey:
264264
'upsample_nearest3d',
265265
'upsample_nearest3d_backward_out',
266266
'upsample_nearest3d_backward',
267-
'upsample_nearest2d_out',
268-
'upsample_nearest2d',
269-
'upsample_nearest2d_backward_out',
270-
'upsample_nearest2d_backward',
267+
# 'upsample_nearest2d_out',
268+
# 'upsample_nearest2d',
269+
# 'upsample_nearest2d_backward_out',
270+
# 'upsample_nearest2d_backward',
271271
'upsample_nearest1d_out',
272272
'upsample_nearest1d',
273273
'upsample_nearest1d_backward_out',

0 commit comments

Comments
 (0)