From 16678bb998ace1b38d02f7c83c117d37ec924220 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 30 Aug 2021 14:00:44 +0800 Subject: [PATCH] fix(dnn): fix_short_cutlass_name_gemm GitOrigin-RevId: cc0a2db9da512a00ad46a493dff8840bfbf2a1d2 --- dnn/scripts/cutlass_generator/gemm_operation.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/dnn/scripts/cutlass_generator/gemm_operation.py b/dnn/scripts/cutlass_generator/gemm_operation.py index 382d6aaa..47a5a82f 100644 --- a/dnn/scripts/cutlass_generator/gemm_operation.py +++ b/dnn/scripts/cutlass_generator/gemm_operation.py @@ -1073,7 +1073,11 @@ void initialize_${operation_name}(Manifest &manifest) { """ # def __enter__(self): - self.kernel_path = os.path.join(self.kernel_path, "%s.cu" % self.operation.procedural_name()) + if self.short_path: + self.kernel_path = os.path.join(self.kernel_path, "%s.cu" % GlobalCnt.cnt) + GlobalCnt.cnt += 1 + else: + self.kernel_path = os.path.join(self.kernel_path, "%s.cu" % self.operation.procedural_name()) self.kernel_file = open(self.kernel_path, "w") self.kernel_file.write(SubstituteTemplate(self.header_template, { 'required_cuda_ver_major': str(self.operation.required_cuda_ver_major),