@Namespace(value="c10::cuda") @NoOffset @Properties(inherit=torch_cuda.class) public class CUDAKernelLaunchRegistry extends Pointer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
CUDAKernelLaunchRegistry() |
CUDAKernelLaunchRegistry(long size)
Native array allocator.
|
CUDAKernelLaunchRegistry(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
boolean |
do_all_devices_support_managed_memory()
If not all devices support DSA, we disable it
|
boolean |
enabled_at_compile_time() |
boolean |
enabled_at_runtime()
Whether or not host-side DSA is enabled or disabled at run-time
Note: Device-side code cannot be enabled/disabled at run-time
|
CUDAKernelLaunchRegistry |
enabled_at_runtime(boolean setter) |
boolean |
gather_launch_stacktrace()
Whether or not to gather stack traces when launching kernels
|
CUDAKernelLaunchRegistry |
gather_launch_stacktrace(boolean setter) |
static CUDAKernelLaunchRegistry |
get_singleton_ref()
Gets the global singleton of the registry
|
DeviceAssertionsData |
get_uvm_assertions_ptr_for_current_device()
Get a pointer to the current device's assertion failure buffer.
|
CUDAKernelLaunchRegistry |
getPointer(long i) |
boolean |
has_failed()
Whether or not a device has indicated a failure
|
int |
insert(BytePointer launch_filename,
BytePointer launch_function,
int launch_linenum,
BytePointer kernel_name,
int stream_id)
Register a new kernel launch and obtain a generation number back to be
passed to the kernel
|
int |
insert(String launch_filename,
String launch_function,
int launch_linenum,
String kernel_name,
int stream_id) |
CUDAKernelLaunchRegistry |
position(long position) |
DeviceAssertionsDataVectorCUDAKernelLaunchInfoVectorPair |
snapshot()
Get copies of the kernel launch registry and each device's assertion
failure buffer so they can be inspected without raising race conditions
|
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public CUDAKernelLaunchRegistry(Pointer p)
Pointer(Pointer)
.public CUDAKernelLaunchRegistry(long size)
Pointer.position(long)
.public CUDAKernelLaunchRegistry()
public CUDAKernelLaunchRegistry position(long position)
public CUDAKernelLaunchRegistry getPointer(long i)
getPointer
in class Pointer
@Cast(value="uint32_t") public int insert(@Cast(value="const char*") BytePointer launch_filename, @Cast(value="const char*") BytePointer launch_function, @Cast(value="const uint32_t") int launch_linenum, @Cast(value="const char*") BytePointer kernel_name, int stream_id)
@Cast(value="uint32_t") public int insert(String launch_filename, String launch_function, @Cast(value="const uint32_t") int launch_linenum, String kernel_name, int stream_id)
@ByVal public DeviceAssertionsDataVectorCUDAKernelLaunchInfoVectorPair snapshot()
public DeviceAssertionsData get_uvm_assertions_ptr_for_current_device()
@ByRef public static CUDAKernelLaunchRegistry get_singleton_ref()
@MemberGetter @Cast(value="const bool") public boolean do_all_devices_support_managed_memory()
@Cast(value="bool") public boolean gather_launch_stacktrace()
public CUDAKernelLaunchRegistry gather_launch_stacktrace(boolean setter)
@Cast(value="bool") public boolean enabled_at_runtime()
public CUDAKernelLaunchRegistry enabled_at_runtime(boolean setter)
@Cast(value="bool") public boolean has_failed()
@MemberGetter @Cast(value="const bool") public boolean enabled_at_compile_time()
Copyright © 2024. All rights reserved.