Skip to content
This repository has been archived by the owner on Oct 11, 2024. It is now read-only.

Commit

Permalink
punica is not needed
Browse files Browse the repository at this point in the history
  • Loading branch information
dhuangnm authored and dhuangnm committed Jun 12, 2024
1 parent e4c1d6f commit 84e1f1c
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions csrc/punica/punica_pybind.cpp
Original file line number Diff line number Diff line change
@@ -1,15 +1,11 @@
#include <torch/extension.h>
#include <git.h>
#include "punica_ops.h"

//====== pybind ======

#define DEFINE_pybind(name) m.def(#name, &name, #name);

std::string githash() { return std::string{git::CommitSHA1()}; }

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def("githash", &githash, "nm-vllm git hash");
m.def("dispatch_bgmv", &dispatch_bgmv, "dispatch_bgmv");
m.def("dispatch_bgmv_low_level", &dispatch_bgmv_low_level,
"dispatch_bgmv_low_level");
Expand Down

0 comments on commit 84e1f1c

Please sign in to comment.