Skip to content

Commit 68a7dc6

Browse files
committed
remove test_cudagraph on win32
1 parent 20a2db6 commit 68a7dc6

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

test/dygraph_to_static/CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@ set(SOT_ENVS SOT_LOG_LEVEL=0 MIN_GRAPH_SIZE=0 STRICT_MODE=False
99
# swgu98: Temporarily commented on Windows platform
1010
if(WIN32)
1111
list(REMOVE_ITEM TEST_OPS test_for_enumerate)
12+
# CUDAGraph is temporarily not supported on Windows platform
13+
list(REMOVE_ITEM TEST_OPS test_cudagraph)
1214
endif()
1315

1416
if(WIN32 AND NOT WITH_GPU)

test/dygraph_to_static/test_cudagraph.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,6 @@ def capture_run_impl(original_run_impl, inputs, parameters, attrs):
5252
"cuda_graph_state": CUDAGraphState.CAPTURE,
5353
"cuda_graph_dispatch_key": inputs[0].shape[0],
5454
}
55-
print("ptr in capture: ", inputs[0].data_ptr())
5655
outputs = original_run_impl(
5756
inputs, parameters, (prog_attrs, cuda_graph_attrs)
5857
)
@@ -74,9 +73,6 @@ def replay_run_impl(original_run_impl, inputs, parameters, attrs):
7473
assert GLOBAL_GRAPH_WITH_BUFFER is not None
7574
GLOBAL_GRAPH_WITH_BUFFER.set_inputs_buffer(inputs)
7675

77-
print(
78-
"ptr in replay: ", GLOBAL_GRAPH_WITH_BUFFER.get_inputs()[0].data_ptr()
79-
)
8076
_ = original_run_impl(
8177
GLOBAL_GRAPH_WITH_BUFFER.get_inputs(),
8278
parameters,

0 commit comments

Comments
 (0)