@Namespace(value="c10::cuda") @NoOffset @Properties(inherit=torch_cuda.class) public class CUDAStream extends Pointer
Modifier and Type | Class and Description |
---|---|
static class |
CUDAStream.Unchecked |
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
CUDAStream(CUDAStream.Unchecked arg0,
Stream stream)
Construct a CUDAStream from a Stream with no error checking.
|
CUDAStream(int arg0,
Stream stream) |
CUDAStream(Pointer p)
Pointer cast constructor.
|
CUDAStream(Stream stream)
Construct a CUDAStream from a Stream.
|
Modifier and Type | Method and Description |
---|---|
Pointer |
asPointer()
Implicit conversion to cudaStream_t.
|
Stream |
asStream()
Implicit conversion to Stream (a.k.a., forget that the stream is a
CUDA stream).
|
byte |
device_index()
Get the CUDA device index that this stream is associated with.
|
torch.DeviceType |
device_type()
Used to avoid baking in device type explicitly to Python-side API.
|
Device |
device()
Get the full Device that this stream is associated with.
|
boolean |
equals(CUDAStream other) |
long |
id()
Return the stream ID corresponding to this particular stream.
|
boolean |
notEquals(CUDAStream other) |
StreamData3 |
pack3()
Reversibly pack a CUDAStream into a struct representation.
|
static T_IntInt_T |
priority_range() |
int |
priority() |
boolean |
query() |
Pointer |
stream()
Explicit conversion to cudaStream_t.
|
void |
synchronize() |
static CUDAStream |
unpack3(long stream_id,
byte device_index,
torch.DeviceType device_type) |
Stream |
unwrap()
Explicit conversion to Stream.
|
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public CUDAStream(Pointer p)
Pointer(Pointer)
.public CUDAStream(@ByVal Stream stream)
public CUDAStream(CUDAStream.Unchecked arg0, @ByVal Stream stream)
@Cast(value="bool") @Name(value="operator ==") @NoException(value=true) public boolean equals(@Const @ByRef CUDAStream other)
@Cast(value="bool") @Name(value="operator !=") @NoException(value=true) public boolean notEquals(@Const @ByRef CUDAStream other)
@Cast(value="cudaStream_t") @Name(value="operator cudaStream_t") public Pointer asPointer()
@ByVal @Name(value="operator c10::Stream") public Stream asStream()
@ByVal public torch.DeviceType device_type()
public byte device_index()
@ByVal public Device device()
public long id()
public void synchronize()
public int priority()
@ByVal public StreamData3 pack3()
@ByVal public static CUDAStream unpack3(long stream_id, byte device_index, @ByVal torch.DeviceType device_type)
@ByVal public static T_IntInt_T priority_range()
Copyright © 2024. All rights reserved.