[Kernel] Add w8a8 CUTLASS kernels (#4749)

This commit is contained in:
Tyler Michael Smith
2024-05-16 18:32:50 -04:00
committed by GitHub
parent 8435b207af
commit 2060e93659
10 changed files with 1197 additions and 2 deletions

View File

@@ -0,0 +1,12 @@
#pragma once
#include "cutlass/cutlass.h"
/**
* Helper function for checking CUTLASS errors
*/
#define CUTLASS_CHECK(status) \
{ \
TORCH_CHECK(status == cutlass::Status::kSuccess, \
cutlassGetStatusString(status)) \
}