@Namespace(value="at") @Properties(inherit=torch.class) public class CUDAHooksInterface extends Pointer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
CUDAHooksInterface()
Default native constructor.
|
CUDAHooksInterface(long size)
Native array allocator.
|
CUDAHooksInterface(Pointer p)
Pointer cast constructor.
|
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 CUDAHooksInterface()
public CUDAHooksInterface(long size)
Pointer.position(long)
.public CUDAHooksInterface(Pointer p)
Pointer(Pointer)
.public CUDAHooksInterface position(long position)
public CUDAHooksInterface getPointer(long i)
getPointer
in class Pointer
public void initCUDA()
@Const @ByRef public Generator getDefaultCUDAGenerator(@Cast(value="c10::DeviceIndex") byte device_index)
@Cast(value="bool") public boolean hasPrimaryContext(@Cast(value="c10::DeviceIndex") byte device_index)
public Allocator getPinnedMemoryAllocator()
public Allocator getCUDADeviceAllocator()
@Cast(value="bool") public boolean supportsDilatedConvolutionWithCuDNN()
@Cast(value="bool") public boolean supportsDepthwiseConvolutionWithCuDNN()
@Cast(value="bool") public boolean supportsBFloat16ConvolutionWithCuDNNv8()
public long versionCuDNN()
public long versionCUDART()
@StdString public BytePointer showConfig()
public double batchnormMinEpsilonCuDNN()
@Cast(value="int64_t") public long cuFFTGetPlanCacheMaxSize(@Cast(value="c10::DeviceIndex") byte arg0)
public void cuFFTSetPlanCacheMaxSize(@Cast(value="c10::DeviceIndex") byte arg0, @Cast(value="int64_t") long arg1)
@Cast(value="int64_t") public long cuFFTGetPlanCacheSize(@Cast(value="c10::DeviceIndex") byte arg0)
public int getNumGPUs()
Copyright © 2024. All rights reserved.