del update_inputs.cu (#10022)
This commit is contained in:
@@ -12,15 +12,19 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from paddle.utils.cpp_extension import CUDAExtension, setup
|
||||
import subprocess
|
||||
|
||||
from paddle.utils.cpp_extension import CUDAExtension, setup
|
||||
|
||||
|
||||
def update_git_submodule():
|
||||
try:
|
||||
subprocess.run(["git", "submodule", "update", "--init"], check=True)
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"Error occurred while updating git submodule: {str(e)}")
|
||||
raise
|
||||
|
||||
|
||||
update_git_submodule()
|
||||
setup(
|
||||
name="paddlenlp_ops",
|
||||
@@ -42,7 +46,6 @@ setup(
|
||||
"./gpu/rebuild_padding_v2.cu",
|
||||
"./gpu/set_value_by_flags_v2.cu",
|
||||
"./gpu/stop_generation_multi_ends_v2.cu",
|
||||
"./gpu/update_inputs.cu",
|
||||
"./gpu/get_output.cc",
|
||||
"./gpu/save_with_output_msg.cc",
|
||||
"./gpu/write_int8_cache_kv.cu",
|
||||
|
||||
Reference in New Issue
Block a user