Skip to content

Commit

Permalink
Building on Ubuntu 24.04 (#129)
Browse files Browse the repository at this point in the history
* build: adding Ubuntu 24.04 to CMakeLists.txt

* build: set OpenVINO version

* fix: **kwargs
  • Loading branch information
ytxmobile98 authored Nov 14, 2024
1 parent 9febeb0 commit 0a449c5
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 7 deletions.
8 changes: 3 additions & 5 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,8 @@ function(get_linux_lsb_release_information)
set(LSB_RELEASE_VERSION "${LSB_RELEASE_VERSION}" PARENT_SCOPE)
endfunction()

set(OV_VERSION_SHORT "nightly")
set(OV_VERSION "2024.3.0.dev20240524_x86_64")
set(OV_VERSION_SHORT "2024.4")
set(OV_VERSION "2024.4.0.16579.c3152d32c9c_x86_64")
set(OV_STORAGE_URL "https://storage.openvinotoolkit.org/repositories/openvino/packages")
set(OV_NIGHTLY_COMMIT "2024.3.0-15502-66093834e38")

Expand All @@ -55,7 +55,7 @@ elseif(UNIX)
if(NOT OV_LIBRARY_URL)
get_linux_lsb_release_information()
if (LSB_RELEASE_ID STREQUAL "Ubuntu")
if (${LSB_RELEASE_VERSION} STREQUAL "18.04" OR ${LSB_RELEASE_VERSION} STREQUAL "20.04" OR ${LSB_RELEASE_VERSION} STREQUAL "22.04")
if (${LSB_RELEASE_VERSION} STREQUAL "18.04" OR ${LSB_RELEASE_VERSION} STREQUAL "20.04" OR ${LSB_RELEASE_VERSION} STREQUAL "22.04" OR ${LSB_RELEASE_VERSION} STREQUAL "24.04")
string(REPLACE ".04" "" LSB_RELEASE_VERSION_SHORT ${LSB_RELEASE_VERSION})
if (${OV_VERSION_SHORT} STREQUAL "nightly")
set(OV_PLATFORM "${OV_NIGHTLY_COMMIT}")
Expand Down Expand Up @@ -115,5 +115,3 @@ target_link_libraries(intel_npu_acceleration_library PRIVATE openvino::runtime)
if (UNIX)
set_target_properties(intel_npu_acceleration_library PROPERTIES LINK_FLAGS "-Wl,-rpath,./")
endif (UNIX)


4 changes: 2 additions & 2 deletions intel_npu_acceleration_library/device.py
Original file line number Diff line number Diff line change
Expand Up @@ -185,7 +185,7 @@ def _parse_to(super_fn: Any, *args: Any, **kwarg: Any):
"""
npu_device, args, kwargs = parse_to_arguments(*args, **kwarg)

device, dtype, non_blocking, convert_to_format = super_fn(*args, *kwargs)
device, dtype, non_blocking, convert_to_format = super_fn(*args, **kwargs)

if npu_device:
device = "npu"
Expand All @@ -204,7 +204,7 @@ def new_to(self, *args: Any, **kwargs: Any):
Returns:
Tensor or Module: The tensor or module with the tensor(s) moved to the specified device(s).
"""
npu_device, args, kwargs = parse_to_arguments(*args, *kwargs)
npu_device, args, kwargs = parse_to_arguments(*args, **kwargs)

if npu_device:
self = convert_to_npu_module(self).to("npu")
Expand Down

0 comments on commit 0a449c5

Please sign in to comment.