Add aten::grid_sampler_3d, aten::grid_sample_3d_backward #898
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Contributor
hjhee
commented
Sep 12, 2024
hjhee
commented
- grid_sampler_3d
- grid_sample_3d_backward
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_bfloat16
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_bool
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_complex128
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_complex64
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_float16
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_int16
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_int32
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_int64
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_int8
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_uint8
test_ops_xpu.py::TestCompositeComplianceXPU::test_forward_ad_nn_functional_grid_sample_xpu_float32
test_ops_xpu.py::TestCommonXPU::test_multiple_devices_nn_functional_grid_sample_xpu_float32
test_nn_xpu.py::TestNNDeviceTypeXPU::test_grid_sample_large_index_3d_xpu_float32
test_nn_xpu.py::TestNNDeviceTypeXPU::test_grid_sample_large_index_3d_xpu_float64
test_ops_fwd_gradients_xpu.py::TestFwdGradientsXPU::test_inplace_forward_mode_AD_nn_functional_grid_sample_xpu_float64
test_ops_gradients_xpu.py::TestBwdGradientsXPU::test_fn_gradgrad_nn_functional_grid_sample_xpu_float64
test_ops_gradients_xpu.py::TestBwdGradientsXPU::test_inplace_grad_nn_functional_grid_sample_xpu_float64
test_ops_gradients_xpu.py::TestBwdGradientsXPU::test_inplace_gradgrad_nn_functional_grid_sample_xpu_float64
test_meta_xpu.py::TestMetaXPU::test_dispatch_meta_inplace_nn_functional_grid_sample_xpu_bfloat16
test_meta_xpu.py::TestMetaXPU::test_dispatch_meta_inplace_nn_functional_grid_sample_xpu_float16
test_meta_xpu.py::TestMetaXPU::test_dispatch_meta_inplace_nn_functional_grid_sample_xpu_float32
test_meta_xpu.py::TestMetaXPU::test_dispatch_meta_inplace_nn_functional_grid_sample_xpu_float64
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_inplace_all_strides_nn_functional_grid_sample_xpu_float32
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_inplace_nn_functional_grid_sample_xpu_bfloat16
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_inplace_nn_functional_grid_sample_xpu_float16
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_inplace_nn_functional_grid_sample_xpu_float32
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_inplace_nn_functional_grid_sample_xpu_float64
test_meta_xpu.py::TestMetaXPU::test_meta_inplace_nn_functional_grid_sample_xpu_bfloat16
test_meta_xpu.py::TestMetaXPU::test_meta_inplace_nn_functional_grid_sample_xpu_float16
test_meta_xpu.py::TestMetaXPU::test_meta_inplace_nn_functional_grid_sample_xpu_float32
test_meta_xpu.py::TestMetaXPU::test_meta_inplace_nn_functional_grid_sample_xpu_float64
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_float32
test_ops_xpu.py::TestCompositeComplianceXPU::test_backward_nn_functional_grid_sample_xpu_float32
test_ops_xpu.py::TestCompositeComplianceXPU::test_operator_nn_functional_grid_sample_xpu_float32
test_ops_xpu.py::TestCompositeComplianceXPU::test_view_replay_nn_functional_grid_sample_xpu_float32
test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_grid_sample_xpu_float32
test_ops_xpu.py::TestCommonXPU::test_dtypes_nn_functional_grid_sample_xpu
test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_nn_functional_grid_sample_xpu_float32
test_ops_xpu.py::TestCommonXPU::test_out_nn_functional_grid_sample_xpu_float32
test_ops_xpu.py::TestCommonXPU::test_out_warning_nn_functional_grid_sample_xpu
test_ops_xpu.py::TestCommonXPU::test_variant_consistency_eager_nn_functional_grid_sample_xpu_float32
test_ops_xpu.py::TestMathBitsXPU::test_neg_view_nn_functional_grid_sample_xpu_float64
test_nn_xpu.py::TestNN::test_grid_sample_3d
test_nn_xpu.py::TestNN::test_grid_sample_error_checking
test_nn_xpu.py::TestNN::test_grid_sample_nearest_neighbor_rounding_mode_consistency
test_nn_xpu.py::TestNNDeviceTypeXPU::test_grid_sample_bfloat16_precision_xpu
test_nn_xpu.py::TestNNDeviceTypeXPU::test_grid_sample_half_precision_xpu
test_nn_xpu.py::TestNNDeviceTypeXPU::test_grid_sample_large_xpu
test_nn_xpu.py::TestNNDeviceTypeXPU::test_grid_sample_nan_inf_xpu_float32
test_nn_xpu.py::TestNNDeviceTypeXPU::test_grid_sample_nan_inf_xpu_float64
test_ops_fwd_gradients_xpu.py::TestFwdGradientsXPU::test_fn_fwgrad_bwgrad_nn_functional_grid_sample_xpu_float64
test_ops_fwd_gradients_xpu.py::TestFwdGradientsXPU::test_forward_mode_AD_nn_functional_grid_sample_xpu_float64
test_ops_gradients_xpu.py::TestBwdGradientsXPU::test_fn_fail_gradgrad_nn_functional_grid_sample_xpu_float64
test_ops_gradients_xpu.py::TestBwdGradientsXPU::test_fn_grad_nn_functional_grid_sample_xpu_float64
test_meta_xpu.py::TestMetaXPU::test_dispatch_meta_outplace_nn_functional_grid_sample_xpu_bfloat16
test_meta_xpu.py::TestMetaXPU::test_dispatch_meta_outplace_nn_functional_grid_sample_xpu_float16
test_meta_xpu.py::TestMetaXPU::test_dispatch_meta_outplace_nn_functional_grid_sample_xpu_float32
test_meta_xpu.py::TestMetaXPU::test_dispatch_meta_outplace_nn_functional_grid_sample_xpu_float64
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_outplace_all_strides_nn_functional_grid_sample_xpu_float32
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_outplace_nn_functional_grid_sample_xpu_bfloat16
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_outplace_nn_functional_grid_sample_xpu_float16
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_outplace_nn_functional_grid_sample_xpu_float32
test_meta_xpu.py::TestMetaXPU::test_dispatch_symbolic_meta_outplace_nn_functional_grid_sample_xpu_float64
test_meta_xpu.py::TestMetaXPU::test_meta_outplace_nn_functional_grid_sample_xpu_bfloat16
test_meta_xpu.py::TestMetaXPU::test_meta_outplace_nn_functional_grid_sample_xpu_float16
test_meta_xpu.py::TestMetaXPU::test_meta_outplace_nn_functional_grid_sample_xpu_float32
test_meta_xpu.py::TestMetaXPU::test_meta_outplace_nn_functional_grid_sample_xpu_float64
|
ZhiweiYan-96
force-pushed
the
hjhee/grid_sampler_3d
branch
from
2e4a998
to
b2ce345
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.