Searched refs:can_vectorize_up_to (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/aten/src/ATen/test/ |
| D | cuda_vectorized_test.cu | 50 ASSERT_EQ(memory::can_vectorize_up_to<bool>(ptr), 4); in TEST()
51 ASSERT_EQ(memory::can_vectorize_up_to<int8_t>(ptr), 4); in TEST()
52 ASSERT_EQ(memory::can_vectorize_up_to<int16_t>(ptr), 4); in TEST()
53 ASSERT_EQ(memory::can_vectorize_up_to<int>(ptr), 4); in TEST()
54 ASSERT_EQ(memory::can_vectorize_up_to<int64_t>(ptr), 4); in TEST()
56 ASSERT_EQ(memory::can_vectorize_up_to<bool>(ptr + 1), 1); in TEST()
57 ASSERT_EQ(memory::can_vectorize_up_to<int8_t>(ptr + 1), 1); in TEST()
59 ASSERT_EQ(memory::can_vectorize_up_to<bool>(ptr + 2), 2); in TEST()
60 ASSERT_EQ(memory::can_vectorize_up_to<int8_t>(ptr + 2), 2); in TEST()
61 ASSERT_EQ(memory::can_vectorize_up_to<int16_t>(ptr + 2), 1); in TEST()
[all …]
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | jit_utils.h | 57 inline int can_vectorize_up_to(size_t default_alignment, void *pointer) { in can_vectorize_up_to() function
68 inline int can_vectorize_up_to(const KernelDescriptor &desc, c10::ArrayRef<char*> pointers) { in can_vectorize_up_to() function
74 int result = can_vectorize_up_to(result_size, pointers[0]); in can_vectorize_up_to()
79 result = std::min(result, can_vectorize_up_to(input_size, pointers[i])); in can_vectorize_up_to()
|
| D | MemoryAccess.cuh | 349 inline C10_HOST_DEVICE int can_vectorize_up_to(const char *pointer) { in can_vectorize_up_to() function
362 inline C10_HOST_DEVICE int can_vectorize_up_to(char *pointer) { in can_vectorize_up_to() function
363 return can_vectorize_up_to<scalar_t>(static_cast<const char*>(pointer)); in can_vectorize_up_to()
373 result = std::min<int>(result, can_vectorize_up_to<arg_t>(pointers[i + 1])); in apply()
378 inline int can_vectorize_up_to(array_t pointers) { in can_vectorize_up_to() function
382 int result = can_vectorize_up_to<return_t>(pointers[0]); in can_vectorize_up_to()
|
| D | CUDAJitLoops.cuh | 120 const int vec_size = at::cuda::jit::can_vectorize_up_to( in launch_jitted_vectorized_kernel()
|
| D | CUDALoops.cuh | 116 int vec_size = memory::can_vectorize_up_to<func_t>(data); in launch_vectorized_kernel()
|
| D | Dropout.cu | 208 vec_size = memory::can_vectorize_up_to<scalar_t>((const char*)self.const_data_ptr()); in get_vector_size()
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | jiterator_impl.h | 47 int can_vectorize_up_to(at::ScalarType type, char* pointer) { in can_vectorize_up_to() function
50 case ScalarType::scalartype : return memory::can_vectorize_up_to<ctype>(pointer); in can_vectorize_up_to()
66 int result = can_vectorize_up_to(result_dtype, static_cast<char*>(iter.data_ptr(0))); in jitted_can_vectorize_up_to()
70 …result = std::min<int>(result, can_vectorize_up_to(common_dtype, static_cast<char*>(iter.data_ptr(… in jitted_can_vectorize_up_to()
|