Skip to content

Commit cf2a43d

Browse files
add_grid_sampler_into_black_list (#19)
1 parent 2b8cf28 commit cf2a43d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torch_ipex/csrc/autocast_mode.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,7 @@ MAKE_REGISTER_FUNC(ADD_NS(reflection_pad1d), "reflection_pad1d", Tensor (const T
203203
MAKE_REGISTER_FUNC(ADD_NS(std), "std", Tensor (const Tensor &, bool), fp32)
204204
MAKE_REGISTER_FUNC(ADD_NS(std), "std.dim", Tensor (const Tensor &, IntArrayRef, bool, bool), fp32)
205205
MAKE_REGISTER_FUNC(ADD_NS(instance_norm), "instance_norm", Tensor (const Tensor &, const c10::optional<Tensor>&, const c10::optional<Tensor>&, const c10::optional<Tensor>&, const c10::optional<Tensor>&, bool, double, double, bool), fp32)
206+
MAKE_REGISTER_FUNC(ADD_NS(grid_sampler), "grid_sampler", Tensor (const Tensor &, const Tensor &, int64_t, int64_t, bool), fp32)
206207

207208
// promote
208209
MAKE_REGISTER_FUNC(ADD_NS(cat), "cat", Tensor (TensorList, int64_t), promote)

0 commit comments

Comments
 (0)