[CI] Some fixes for the build (#451)

This commit is contained in:
Philippe Tillet
2022-02-06 19:11:33 -08:00
committed by GitHub
parent 5a8a544d10
commit 7b48340ffd
2 changed files with 1 additions and 3 deletions

View File

@@ -150,7 +150,7 @@ void parse_args(py::list& args, py::list do_not_specialize, const std::string& f
for(int i = 0; i < len; i++){
cache_key += "_";
py::int_ py_i = py::int_(i);
bool specialize = std::find(do_not_specialize.begin(), do_not_specialize.end(), py_i) == do_not_specialize.end();
bool specialize = !do_not_specialize.contains(py_i);
py::object arg = args[i];
auto arg_ptr = arg.ptr();

View File

@@ -169,8 +169,6 @@ import triton.language as tl
],
key=['M', 'N', 'K'],
)
# %
# We can now define our kernel as normal, using all the techniques presented above
@triton.jit
def matmul_kernel(
# Pointers to matrices