MCPcopy Create free account
hub / github.com/pytorch/pytorch / current_stream

Method current_stream

torch/_dynamo/device_interface.py:85–86  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

83
84 @staticmethod
85 def current_stream():
86 raise NotImplementedError()
87
88 @staticmethod
89 def set_stream(stream: torch.Stream):

Callers 15

__dlpack__Method · 0.80
from_dlpackFunction · 0.80
to_mapFunction · 0.80
__init__Method · 0.80
__init__Method · 0.80
_init_streamsFunction · 0.80
_pre_forward_unshardFunction · 0.80
_root_pre_forwardFunction · 0.80
_pre_backward_hookFunction · 0.80
_post_backward_hookFunction · 0.80
_cast_grad_to_param_dtypeFunction · 0.80
_post_backward_final_callbackFunction · 0.80

Calls

no outgoing calls

Tested by 15

async_cuda_sleep_and_set_to_oneFunction · 0.64
forwardMethod · 0.64
_slow_add_on_user_streamMethod · 0.64
test_fused_attention_vs_math_ref_grads_cudagraphMethod · 0.64
test_current_streamMethod · 0.64
test_default_streamMethod · 0.64
test_stream_event_deviceMethod · 0.64
test_stream_contextMethod · 0.64
test_streams_multi_gpuMethod · 0.64
test_streams_multi_gpu_queryMethod · 0.64
test_streams_multi_gpu_eqMethod · 0.64
_stream_synchronizeMethod · 0.64