Change the clean shader API to clean local data

To be align with the unified API.

Change-Id: I2819339fba6f528204cebd3e9605109e82cbc5b4
此提交包含在:
Bill(Shuzhou) Liu committed 2024-06-17 16:23:33 -05:00
1 父節點 94cab382bf
當前提交 e3c63628e5
10 file changed
+31 -39

No files matched your search

+3 -4
查看文件
@@ -2084,17 +2084,16 @@ except AmdSmiException as e:
print(e)
```
### amdsmi_set_gpu_run_cleaner_shader
### amdsmi_clean_gpu_local_data
Description: Clear the SRAM data of the given device. This can be called between user logins to prevent information leak.
Input parameters:
* `processor_handle` handle for the given device
* `sclean` the clean flag. Only 1 will take effect and other number are reserved for future usage.
Output: None
Exceptions that can be thrown by `amdsmi_set_gpu_run_cleaner_shader` function:
Exceptions that can be thrown by `amdsmi_clean_gpu_local_data` function:
* `AmdSmiLibraryException`
* `AmdSmiRetryException`
@@ -2109,7 +2108,7 @@ try:
print("No GPUs on machine")
else:
for device in devices:
amdsmi_set_gpu_run_cleaner_shader(device, 1)
amdsmi_clean_gpu_local_data(device)
except AmdSmiException as e:
print(e)
```