提交 16678bb9 编写于 作者: M Megvii Engine Team

fix(dnn): fix_short_cutlass_name_gemm

GitOrigin-RevId: cc0a2db9da512a00ad46a493dff8840bfbf2a1d2
上级 6f455ae9
......@@ -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),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册