llm.c icon indicating copy to clipboard operation
llm.c copied to clipboard

Update the -lcublas -lcublasLt flag in the comment

Open lancerts opened this issue 9 months ago • 0 comments

Due to the void setup_main() in the common.h,

    // setup cuBLAS and cuBLASLt
    cublasCheck(cublasCreate(&cublas_handle));
    cublasCheck(cublasLtCreate(&cublaslt_handle));
    cudaCheck(cudaMalloc(&cublaslt_workspace, cublaslt_workspace_size));

added -lcublas -lcublasLt in the comment for compile for consistency.

Or, we can simply remove the comment for compile since everything is in the makefile already.

lancerts avatar May 10 '24 05:05 lancerts