minor: update header and use pytest (#3054)
This commit is contained in:
@@ -16,7 +16,7 @@
|
||||
#include "cutlass_extensions/epilogue/epilogue_per_row_per_col_scale.h"
|
||||
#include "cutlass_extensions/gemm/gemm_universal_base_compat.h"
|
||||
#include "cutlass_extensions/gemm/gemm_with_epilogue_visitor.h"
|
||||
#include "utils.hpp"
|
||||
#include "utils.h"
|
||||
|
||||
using namespace cute;
|
||||
|
||||
|
||||
@@ -6,7 +6,7 @@
|
||||
|
||||
#include <THC/THCAtomics.cuh>
|
||||
|
||||
#include "utils.hpp"
|
||||
#include "utils.h"
|
||||
|
||||
#ifdef USE_ROCM
|
||||
#include <hip/hip_runtime.h>
|
||||
|
||||
@@ -4,7 +4,7 @@
|
||||
|
||||
#include <THC/THCAtomics.cuh>
|
||||
|
||||
#include "utils.hpp"
|
||||
#include "utils.h"
|
||||
#include "vectorization.cuh"
|
||||
|
||||
template <typename scalar_t>
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
#include <vector>
|
||||
|
||||
#include "utils.hpp"
|
||||
#include "utils.h"
|
||||
|
||||
// trt_reduce
|
||||
using fptr_t = int64_t;
|
||||
|
||||
@@ -21,7 +21,7 @@
|
||||
#include <stdint.h>
|
||||
#include <torch/all.h>
|
||||
|
||||
#include "utils.hpp"
|
||||
#include "utils.h"
|
||||
|
||||
namespace trt_llm {
|
||||
constexpr size_t WARP_SIZE = 32;
|
||||
|
||||
Reference in New Issue
Block a user