Skip to content

Commit

Permalink
clean rocm
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz committed Sep 16, 2023
1 parent 25c25be commit 27d8e5f
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions source/op/prod_env_mat_multi_device.cc
Original file line number Diff line number Diff line change
Expand Up @@ -669,7 +669,6 @@ class ProdEnvMatAOp : public OpKernel {
int* type_cpy;
int frame_nall = nall;
int mesh_tensor_size = static_cast<int>(mesh_tensor.NumElements());
std::vector<Tensor> tensor_list(7);
// prepare coord and nlist
_prepare_coord_nlist_gpu_rocm<FPTYPE>(
context, &tensor_list[0], &coord, coord_cpy, &type, type_cpy,
Expand Down Expand Up @@ -994,7 +993,6 @@ class ProdEnvMatROp : public OpKernel {
int* type_cpy;
int frame_nall = nall;
int mesh_tensor_size = static_cast<int>(mesh_tensor.NumElements());
std::vector<Tensor> tensor_list(7);
// prepare coord and nlist
_prepare_coord_nlist_gpu_rocm<FPTYPE>(
context, &tensor_list[0], &coord, coord_cpy, &type, type_cpy,
Expand Down Expand Up @@ -1371,7 +1369,6 @@ class ProdEnvMatAMixOp : public OpKernel {
int* type_cpy;
int frame_nall = nall;
int mesh_tensor_size = static_cast<int>(mesh_tensor.NumElements());
std::vector<Tensor> tensor_list(7);
// prepare coord and nlist
_prepare_coord_nlist_gpu_rocm<FPTYPE>(
context, &tensor_list[0], &coord, coord_cpy, &f_type, type_cpy,
Expand Down

0 comments on commit 27d8e5f

Please sign in to comment.