From 5620f8e601ac8554d6a9a53a2e51b0384be0d3ae Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Wed, 13 Jul 2022 16:36:29 +0800
Subject: [PATCH] fix issue

---
 templates/repo/modelarts/inferencejob/index.tmpl | 34 ++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/templates/repo/modelarts/inferencejob/index.tmpl b/templates/repo/modelarts/inferencejob/index.tmpl
index e22cca820..1dde32469 100644
--- a/templates/repo/modelarts/inferencejob/index.tmpl
+++ b/templates/repo/modelarts/inferencejob/index.tmpl
@@ -39,6 +39,16 @@
                     </div>
                 </div>
                 <div class="column right aligned">
+                    <div class="ui selection dropdown" style="min-width: 10em;min-height:2.6em;border-radius: .28571429rem;margin-right: 1em;padding: .67em 3.2em .7em 1em;">
+                        {{svg "octicon-server" 16}}
+                        <div class="default text" style="color: rgba(0,0,0,.87);"></div>
+                        <i class="dropdown icon"></i>
+                        <div class="menu">
+                            <div class="item" data-value="all">{{$.i18n.Tr "repo.gpu_type_all"}}</div>
+                            <div class="item"  data-value="CPU/GPU">CPU/GPU</div>
+                            <div class="item"  data-value="NPU">NPU</div>
+                        </div>
+                    </div>
                     {{if .Permission.CanWrite $.UnitTypeCloudBrain}}
                     <a class="ui green button" href="{{.RepoLink}}/cloudbrain/inference-job/create">{{$.i18n.Tr "repo.modelarts.train_job.new_infer"}}</a>
                     {{else}}
@@ -264,3 +274,27 @@
 </div>
 {{template "base/footer" .}}
 
+<script>
+    let url = {{$.RepoLink}};
+    let all = {{$.i18n.Tr "repo.gpu_type_all"}}
+    $(document).ready(function(){
+    const params = new URLSearchParams(location.search)
+    if(!location.search){
+        $('.default.text').text(all)
+    }else{
+        if(!params.has('listType') || params.get('listType')=='all'){
+            $('.default.text').text(all)
+        }
+        else{
+            $('.default.text').text(params.get('listType'))
+        }
+    }
+    
+    $('.ui.selection.dropdown').dropdown({
+        onChange:function(value){
+            location.href = `${url}/modelarts/inference-job?listType=${value}`
+        }
+        })
+    })
+</script>
+