Skip to content

Commit 81ccb8e

Browse files
authored
Update dgemm_vbatch.cu
1 parent 204f279 commit 81ccb8e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

source/source_lcao/module_gint/temp_gint/kernel/dgemm_vbatch.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
#include "gemm_tn_vbatch.cuh"
22
#include "gemm_nn_vbatch.cuh"
33
#include "dgemm_vbatch.h"
4+
#include "source_base/module_device/device.h"
45

56
void dgemm_nn_vbatch(
67
int max_m, int max_n, int max_k,

0 commit comments

Comments
 (0)