Browse Source

fix(dnn): fix_short_cutlass_name_gemm

GitOrigin-RevId: cc0a2db9da
release-1.6
Megvii Engine Team 3 years ago
parent
commit
16678bb998
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      dnn/scripts/cutlass_generator/gemm_operation.py

+ 5
- 1
dnn/scripts/cutlass_generator/gemm_operation.py View File

@@ -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),


Loading…
Cancel
Save