mcst-linux-kernel/patches-2024.06.26/pytorch-1.13.1/0007-cuda-val-undefined-ref...

38 lines
1.0 KiB
Diff

Subject: stub for undefined reference to '_ZTIN5torch3jit5fuser4cuda3ValE'
Bug: 152612
Tags: lcc
diff -rupN a/torch/csrc/jit/codegen/cuda/ir_internal_nodes.h b/torch/csrc/jit/codegen/cuda/ir_internal_nodes.h
--- a/torch/csrc/jit/codegen/cuda/ir_internal_nodes.h 2023-10-05 17:48:59.420587857 +0300
+++ b/torch/csrc/jit/codegen/cuda/ir_internal_nodes.h 2023-10-06 02:11:46.689142630 +0300
@@ -353,7 +353,9 @@ class TORCH_CUDA_CU_API WelfordTriplet {
}
TensorView* avgTv() const {
+#ifndef __LCC__
TORCH_INTERNAL_ASSERT(avg()->isA<TensorView>());
+#endif
return avg()->as<TensorView>();
}
@@ -366,7 +368,9 @@ class TORCH_CUDA_CU_API WelfordTriplet {
}
TensorView* varTv() const {
+#ifndef __LCC__
TORCH_INTERNAL_ASSERT(var()->isA<TensorView>());
+#endif
return var()->as<TensorView>();
}
@@ -379,7 +383,9 @@ class TORCH_CUDA_CU_API WelfordTriplet {
}
TensorView* NTv() const {
+#ifndef __LCC__
TORCH_INTERNAL_ASSERT(N()->isA<TensorView>());
+#endif
return N()->as<TensorView>();
}