Skip to content

Commit 02bfc16

Browse files
committed
remove printing kernel output entirely
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent ad62229 commit 02bfc16

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

utils/SPIRVRunner/SPIRVRunner.cpp

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -406,16 +406,11 @@ bool check_option_amoung_argv(int argc, char **argv, std::string option) {
406406

407407
int main(int argc, char **argv) {
408408
try {
409-
std::string print_output_kernel_tensor = "--print-output-kernel-tensor";
410409
std::string enable_profiling = "--enable-profiling";
411410
if (argc < 2) {
412411
std::cout << "Help: " << std::endl;
413412
std::cout << "<Executable> <Output Tensor Name>" << std::endl;
414413
std::cout << "./build/SPIRVRunner tensor_2" << std::endl;
415-
std::cout << "To print the output kernel tensor to stdout, use:"
416-
<< std::endl;
417-
std::cout << "./build/SPIRVRunner tensor_2 " << print_output_kernel_tensor
418-
<< std::endl;
419414
std::cout << "To get kernel time, use:" << std::endl;
420415
std::cout << "./build/SPIRVRunner tensor_2 " << enable_profiling
421416
<< std::endl;
@@ -455,10 +450,6 @@ int main(int argc, char **argv) {
455450

456451
auto output = launchKernel(q, kernel, tritonArgDict, get_kernel_time);
457452

458-
if (check_option_amoung_argv(argc, argv, print_output_kernel_tensor)) {
459-
std::cout << "Kernel return output: " << output[0] << std::endl;
460-
}
461-
462453
auto output_tensor = tritonArgDict.spirv_dump_dir + "/cpp_outs.pt";
463454
write_tensor(output_tensor, output);
464455
std::cout << "Output Tensor Path: " << output_tensor << std::endl;

0 commit comments

Comments
 (0)