scatter cannot be used w/ long*

[saadrahim]

Compile time error from PyTorch:

In file included from /pytorch/aten/src/ATen/native/hip/Unique.hip:5:
In file included from /opt/rocm/include/thrust/execution_policy.h:31:
In file included from /opt/rocm/include/thrust/system/cpp/execution_policy.h:64:
In file included from /opt/rocm/include/thrust/system/cpp/detail/sort.h:22:
In file included from /opt/rocm/include/thrust/system/detail/sequential/sort.h:63:
In file included from /opt/rocm/include/thrust/system/detail/sequential/sort.inl:23:
In file included from /opt/rocm/include/thrust/system/detail/sequential/stable_primitive_sort.h:55:
In file included from /opt/rocm/include/thrust/system/detail/sequential/stable_primitive_sort.inl:21:
In file included from /opt/rocm/include/thrust/system/detail/sequential/stable_radix_sort.h:55:
In file included from /opt/rocm/include/thrust/system/detail/sequential/stable_radix_sort.inl:20:
In file included from /opt/rocm/include/thrust/copy.h:513:
In file included from /opt/rocm/include/thrust/detail/copy_if.h:74:
In file included from /opt/rocm/include/thrust/detail/copy_if.inl:20:
In file included from /opt/rocm/include/thrust/system/detail/generic/copy_if.h:63:
In file included from /opt/rocm/include/thrust/system/detail/generic/copy_if.inl:31:
In file included from /opt/rocm/include/thrust/scan.h:1563:
In file included from /opt/rocm/include/thrust/detail/scan.inl:29:
In file included from /opt/rocm/include/thrust/system/detail/adl/scan_by_key.h:44:
In file included from /opt/rocm/include/thrust/system/hip/detail/scan_by_key.h:33:
In file included from /opt/rocm/include/thrust/system/hip/execution_policy.h:41:
In file included from /opt/rocm/include/thrust/system/hip/detail/count.h:35:
In file included from /opt/rocm/include/thrust/system/hip/detail/reduce.h:42:
In file included from /opt/rocm/include/thrust/device_vector.h:26:
In file included from /opt/rocm/include/thrust/detail/vector_base.h:547:
In file included from /opt/rocm/include/thrust/detail/vector_base.inl:25:
In file included from /opt/rocm/include/thrust/equal.h:237:
In file included from /opt/rocm/include/thrust/detail/equal.inl:25:
In file included from /opt/rocm/include/thrust/system/detail/generic/equal.h:47:
In file included from /opt/rocm/include/thrust/system/detail/generic/equal.inl:21:
In file included from /opt/rocm/include/thrust/mismatch.h:259:
In file included from /opt/rocm/include/thrust/detail/mismatch.inl:27:
In file included from /opt/rocm/include/thrust/system/detail/generic/mismatch.h:57:
In file included from /opt/rocm/include/thrust/system/detail/generic/mismatch.inl:21:
In file included from /opt/rocm/include/thrust/find.h:384:
In file included from /opt/rocm/include/thrust/detail/find.inl:25:
In file included from /opt/rocm/include/thrust/system/detail/generic/find.h:62:
In file included from /opt/rocm/include/thrust/system/detail/generic/find.inl:19:
In file included from /opt/rocm/include/thrust/reduce.h:784:
In file included from /opt/rocm/include/thrust/detail/reduce.inl:26:
In file included from /opt/rocm/include/thrust/system/detail/generic/reduce_by_key.h:88:
In file included from /opt/rocm/include/thrust/system/detail/generic/reduce_by_key.inl:30:
In file included from /opt/rocm/include/thrust/scatter.h:422:
In file included from /opt/rocm/include/thrust/detail/scatter.inl:26:
In file included from /opt/rocm/include/thrust/system/detail/adl/scatter.h:44:
/opt/rocm/include/thrust/system/hip/detail/scatter.h:50:20: error: type 'long *' cannot be used prior to '::' because it has no members
typedef typename ItemsIt::value_type ItemsTy;
^
/opt/rocm/include/thrust/detail/scatter.inl:45:10: note: in instantiation of function template specialization 'thrust::hip_rocprim::scatter<thrust::detail::execute_with_allocator<THCThrustAllocator, execute_on_stream_base>, long *, const long *, long *>' requested here
return scatter(thrust::detail::derived_cast(thrust::detail::strip_const(exec)), first, last, map, output);
^
/pytorch/aten/src/ATen/native/hip/Unique.hip:51:13: note: in instantiation of function template specialization 'thrust::scatter<thrust::detail::execute_with_allocator<THCThrustAllocator, execute_on_stream_base>, long *, const long *, long *>' requested here
thrust::scatter(policy, inv_loc_ptr, inv_loc_ptr + num_inp, sorted_indices_ptr, inverse_indices_ptr);
^
/pytorch/aten/src/ATen/native/hip/Unique.hip:105:48: note: in instantiation of function template specialization 'at::native::(anonymous namespace)::compute_unique<thrust::detail::execute_with_allocator<THCThrustAllocator, execute_on_stream_base>, unsigned char, thrust::equal_to, thrust::not_equal_to >' requested here
std::tie(inverse_indices, counts, num_out) = compute_unique(
^
/pytorch/aten/src/ATen/native/hip/Unique.hip:207:46: note: in instantiation of function template specialization 'at::native::(anonymous namespace)::unique_cuda_template' requested here
std::tie(output, inverse, std::ignore) = unique_cuda_template<scalar_t>(self, false, return_inverse, false);
^