Types[Index] = KernelArgumentType::VALUE;
}
- // Pack a GlobalDeviceMemoryBase argument.
- void PackOneArgument(size_t Index, const GlobalDeviceMemoryBase &Argument) {
- Addresses[Index] = Argument.getHandle();
- Sizes[Index] = sizeof(void *);
- Types[Index] = KernelArgumentType::GLOBAL_DEVICE_MEMORY;
- }
-
- // Pack a GlobalDeviceMemoryBase pointer argument.
- void PackOneArgument(size_t Index, GlobalDeviceMemoryBase *Argument) {
- Addresses[Index] = Argument->getHandle();
- Sizes[Index] = sizeof(void *);
- Types[Index] = KernelArgumentType::GLOBAL_DEVICE_MEMORY;
- }
-
- // Pack a const GlobalDeviceMemoryBase pointer argument.
- void PackOneArgument(size_t Index, const GlobalDeviceMemoryBase *Argument) {
- Addresses[Index] = Argument->getHandle();
- Sizes[Index] = sizeof(void *);
- Types[Index] = KernelArgumentType::GLOBAL_DEVICE_MEMORY;
- }
-
// Pack a GlobalDeviceMemory<T> argument.
template <typename T>
void PackOneArgument(size_t Index, const GlobalDeviceMemory<T> &Argument) {
- Addresses[Index] = Argument.getHandle();
+ Addresses[Index] = Argument.getHandleAddress();
Sizes[Index] = sizeof(void *);
Types[Index] = KernelArgumentType::GLOBAL_DEVICE_MEMORY;
}
// Pack a GlobalDeviceMemory<T> pointer argument.
template <typename T>
void PackOneArgument(size_t Index, GlobalDeviceMemory<T> *Argument) {
- Addresses[Index] = Argument->getHandle();
+ Addresses[Index] = Argument->getHandleAddress();
Sizes[Index] = sizeof(void *);
Types[Index] = KernelArgumentType::GLOBAL_DEVICE_MEMORY;
}
// Pack a const GlobalDeviceMemory<T> pointer argument.
template <typename T>
void PackOneArgument(size_t Index, const GlobalDeviceMemory<T> *Argument) {
- Addresses[Index] = Argument->getHandle();
+ Addresses[Index] = Argument->getHandleAddress();
Sizes[Index] = sizeof(void *);
Types[Index] = KernelArgumentType::GLOBAL_DEVICE_MEMORY;
}
TEST_F(DeviceMemoryPackingTest, SingleTypedGlobal) {
auto Array = se::make_kernel_argument_pack(TypedGlobal);
- ExpectEqual(TypedGlobal.getHandle(), sizeof(void *),
+ ExpectEqual(TypedGlobal.getHandleAddress(), sizeof(void *),
Type::GLOBAL_DEVICE_MEMORY, Array, 0);
EXPECT_EQ(1u, Array.getArgumentCount());
EXPECT_EQ(0u, Array.getSharedCount());
TEST_F(DeviceMemoryPackingTest, SingleTypedGlobalPointer) {
auto Array = se::make_kernel_argument_pack(&TypedGlobal);
- ExpectEqual(TypedGlobal.getHandle(), sizeof(void *),
+ ExpectEqual(TypedGlobal.getHandleAddress(), sizeof(void *),
Type::GLOBAL_DEVICE_MEMORY, Array, 0);
EXPECT_EQ(1u, Array.getArgumentCount());
EXPECT_EQ(0u, Array.getSharedCount());
TEST_F(DeviceMemoryPackingTest, SingleConstTypedGlobalPointer) {
const se::GlobalDeviceMemory<int> *ArgumentPointer = &TypedGlobal;
auto Array = se::make_kernel_argument_pack(ArgumentPointer);
- ExpectEqual(TypedGlobal.getHandle(), sizeof(void *),
+ ExpectEqual(TypedGlobal.getHandleAddress(), sizeof(void *),
Type::GLOBAL_DEVICE_MEMORY, Array, 0);
EXPECT_EQ(1u, Array.getArgumentCount());
EXPECT_EQ(0u, Array.getSharedCount());
TypedGlobalPointer, TypedShared,
&TypedShared, TypedSharedPointer);
ExpectEqual(&Value, sizeof(Value), Type::VALUE, Array, 0);
- ExpectEqual(TypedGlobal.getHandle(), sizeof(void *),
+ ExpectEqual(TypedGlobal.getHandleAddress(), sizeof(void *),
Type::GLOBAL_DEVICE_MEMORY, Array, 1);
- ExpectEqual(TypedGlobal.getHandle(), sizeof(void *),
+ ExpectEqual(TypedGlobal.getHandleAddress(), sizeof(void *),
Type::GLOBAL_DEVICE_MEMORY, Array, 2);
- ExpectEqual(TypedGlobal.getHandle(), sizeof(void *),
+ ExpectEqual(TypedGlobal.getHandleAddress(), sizeof(void *),
Type::GLOBAL_DEVICE_MEMORY, Array, 3);
ExpectEqual(nullptr, TypedShared.getByteCount(), Type::SHARED_DEVICE_MEMORY,
Array, 4);