llm.c
llm.c copied to clipboard
Update the -lcublas -lcublasLt flag in the comment
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.